Tpk - UnityVersion - fix fromString

K0lb3 · Nov 19, 2024 · eece6ad · eece6ad
1 parent c477355
commit eece6ad
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/UnityPy/helpers/Tpk.py b/UnityPy/helpers/Tpk.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import re
 from enum import IntEnum, IntFlag
 from importlib.resources import open_binary
 from io import BytesIO
@@ -300,7 +301,11 @@ def fromStream(stream: BytesIO) -> UnityVersion:
 
     @staticmethod
     def fromString(version: str) -> UnityVersion:
-        return UnityVersion(version.split("."))
+        match = re.match(r"(\d+)\.(\d+)\.(\d+)(([a-zA-Z]+)(\d+))?", version)
+        if not match:
+            raise ValueError("Invalid version string")
+        major, minor, patch, _, type, build = match.groups()
+        return UnityVersion.fromList(int(major), int(minor), int(patch), int(build))
 
     @staticmethod
     def fromList(