Merge pull request #1 from iron3oxide/update-typeid-from-uuid-fix

fix: Improve type prefix and suffix extraction
akhundMurad · May 13, 2024 · 71510ca · 71510ca
2 parents 0ead5f1 + c6a4611
commit 71510ca
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/typeid/typeid.py b/typeid/typeid.py
@@ -80,15 +80,12 @@ def from_uuid(suffix: uuid6.UUID, prefix: Optional[str] = None) -> TypeID:
 
 
 def get_prefix_and_suffix(string: str) -> tuple:
-    parts = string.split("_")
-    prefix = None
+    parts = string.rsplit("_", 1)
     if len(parts) == 1:
+        prefix = None
         suffix = parts[0]
-    elif len(parts) == 2 and parts[0] != "":
-        suffix = parts[1]
-        prefix = parts[0]
     else:
-        raise InvalidTypeIDStringException(f"Invalid TypeID: {string}")
+        prefix, suffix = parts
 
     return prefix, suffix