fixed merging issues

Hochfrequenz · Aug 14, 2024 · 2c60871 · 2c60871
1 parent 320f77f
commit 2c60871
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/src/migmose/parsing.py b/src/migmose/parsing.py
@@ -231,3 +231,19 @@ def process_segmentlayouts(segmentlayout_tables: list[list[_Cell]]) -> dict[str,
             )
         )
     return segment_layouts_dict
+
+
+def _extract_document_version(path: Path) -> str:
+    document_str = str(path)
+    pattern = (
+        r"MIG(?:Strom|Gas)?-?informatorischeLesefassung?(.*?)"
+        r"(?:_|KonsolidierteLesefassung|-AußerordentlicheVeröffentlichung)"
+    )
+    matches = re.search(pattern, document_str, re.IGNORECASE)
+    if matches:
+        document_version = matches.group(1)
+        if document_version == "":
+            logger.warning(f"❌ No document version found in {path}.", fg="red")
+        return document_version
+    logger.error(f"❌ Unexpected document name in {path}.", fg="red")
+    return ""
diff --git a/unittests/test_reduced_nested_nachrichtenstruktur.py b/unittests/test_reduced_nested_nachrichtenstruktur.py
@@ -84,7 +84,7 @@ def test_output_tree(self, message_format: EdifactFormat, tmp_path, snapshot):
             [message_format], path_to_test_edi_energy_mirror_repo, EdifactFormatVersion.FV2310
         )
         file_path = file_path_dict[message_format]
-        raw_lines = parse_raw_nachrichtenstrukturzeile(file_path)
+        raw_lines, _ = parse_raw_nachrichtenstrukturzeile(file_path)
         nachrichtenstrukturtabelle = NachrichtenstrukturTabelle.create_nachrichtenstruktur_tabelle(raw_lines)
         nested_nachrichtenstruktur, _ = NestedNachrichtenstruktur.create_nested_nachrichtenstruktur(
             nachrichtenstrukturtabelle

diff --git a/unittests/test_segment_group_hierarchies.py b/unittests/test_segment_group_hierarchies.py
@@ -30,7 +30,7 @@ def test_create_create_segmentgroup_hierarchy(self, message_format: EdifactForma
             [message_format], path_to_test_edi_energy_mirror_repo, EdifactFormatVersion.FV2310
         )
         file_path = file_path_dict[message_format]
-        raw_lines = parse_raw_nachrichtenstrukturzeile(file_path)
+        (raw_lines, _) = parse_raw_nachrichtenstrukturzeile(file_path)
         nachrichtenstrukturtabelle = NachrichtenstrukturTabelle.create_nachrichtenstruktur_tabelle(raw_lines)
         nested_nachrichtenstruktur, _ = NestedNachrichtenstruktur.create_nested_nachrichtenstruktur(
             nachrichtenstrukturtabelle