Patch small bugs (#125)

* fix small bug for all message formats * Changed pattern for nr in NachrichtenstrukturZeile * Update src/migmose/mig/nachrichtenstrukturzeile.py Co-authored-by: konstantin <konstantin.klein@hochfrequenz.de> --------- Co-authored-by: konstantin <konstantin.klein@hochfrequenz.de>
Hochfrequenz · Jul 30, 2024 · 923642a · 923642a
1 parent 0d93de1
commit 923642a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/migmose/__main__.py b/src/migmose/__main__.py
@@ -19,7 +19,7 @@ def check_message_format(ctx, param, value) -> list[EdifactFormat]:  # type: ign
     """
     Check if the message format is valid.
     """
-    if value is None:
+    if len(value) == 0:
         value = map(lambda x: x.name, EdifactFormat)
     return list(value)
 

diff --git a/src/migmose/mig/nachrichtenstrukturzeile.py b/src/migmose/mig/nachrichtenstrukturzeile.py
@@ -24,7 +24,9 @@ class for lines in mig tables, e.g. (ORDCHG):
     """
 
     zaehler: str
-    nr: str | None = Field(default=None, pattern=r"^\d{5}$")  #: the segment ID (can be used to match with AHBs)
+    nr: str | None = Field(
+        default=None, pattern=r"^(?:\d{5})|(?:[A-Za-z]+\d{4})$"
+    )  #: the segment ID (can be used to match with AHBs)
     bezeichnung: str
     standard_status: str
     bdew_status: str