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