From a25ec003e840dc458a89ea8ef6a575625fc2af78 Mon Sep 17 00:00:00 2001
From: Leah Tacke genannt Unterberg <leah.tgu@pads.rwth-aachen.de>
Date: Thu, 8 May 2025 15:50:35 +0200
Subject: [PATCH] fixing unlinted programming erros

---
 mitm_tooling/extraction/sql/data_models/db_probe.py | 2 ++
 mitm_tooling/extraction/sql/mapping/export.py       | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/mitm_tooling/extraction/sql/data_models/db_probe.py b/mitm_tooling/extraction/sql/data_models/db_probe.py
index e8c6a54..3de92f4 100644
--- a/mitm_tooling/extraction/sql/data_models/db_probe.py
+++ b/mitm_tooling/extraction/sql/data_models/db_probe.py
@@ -81,6 +81,8 @@ class DBProbe(DBProbeBase):
     def update_probes(self, *probes: tuple[ShortTableIdentifier, TableProbe]):
         for ti, tp in probes:
             schema_name, table_name = ti
+            if not schema_name in self.db_structured_table_probes:
+                self.db_structured_table_probes[schema_name] = {}
             self.db_structured_table_probes[schema_name][table_name] = tp
 
     def drop_probes(self, *to_drop: ShortTableIdentifier):
diff --git a/mitm_tooling/extraction/sql/mapping/export.py b/mitm_tooling/extraction/sql/mapping/export.py
index 539c188..826d4c1 100644
--- a/mitm_tooling/extraction/sql/mapping/export.py
+++ b/mitm_tooling/extraction/sql/mapping/export.py
@@ -97,7 +97,7 @@ class MappingExport(pydantic.BaseModel):
         data_providers: dict[ConceptName, list[DataProvider]] = {}
 
         meta = sa.MetaData(schema='export')
-        for concept_mapping in self.mapped_concepts:
+        for i, concept_mapping in enumerate(self.mapped_concepts):
             if concept_mapping.mitm != self.mitm:
                 continue
 
-- 
GitLab