util: version: version parsing helper

Add a helper class for parsing version strings and test.

Signed-off-by: Jordan Yates <jordan@embeint.com>

Author: JordanYates

src/infuse_iot/util/version.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env
+
+from typing import Self
+
+
+class Version:
+    def __init__(self, major: int, minor: int, revision: int, build_num: int):
+        self.major = major
+        self.minor = minor
+        self.revision = revision
+        self.build_num = build_num
+
+    @classmethod
+    def from_string(cls, version_string: str) -> Self:
+        "Convert 'x.y.z+rev' string to version instance"
+
+        rev_split = version_string.split("+")
+        if len(rev_split) != 2:
+            raise ValueError(f"'{version_string}' is not a valid version string")
+        ver_split = rev_split[0].split(".")
+        if len(ver_split) != 3:
+            raise ValueError(f"'{version_string}' is not a valid version string")
+        return cls(int(ver_split[0]), int(ver_split[1]), int(ver_split[2]), int(rev_split[1], 16))
+
+    def __str__(self):
+        return f"{self.major}.{self.minor}.{self.revision}+{self.build_num:08x}"
+
+    def __eq__(self, other):
+        if not isinstance(other, Version):
+            raise NotImplementedError
+        return (
+            self.major == other.major
+            and self.minor == other.minor
+            and self.revision == other.revision
+            and self.build_num == other.build_num
+        )

tests/util/test_version.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+
+import os
+
+import pytest
+
+from infuse_iot.util.version import Version
+
+assert "TOXTEMPDIR" in os.environ, "you must run these tests using tox"
+
+
+def test_version_parsing():
+    with pytest.raises(ValueError):
+        Version.from_string("random string")
+    with pytest.raises(ValueError):
+        Version.from_string("1.2.3")
+    with pytest.raises(ValueError):
+        Version.from_string("1.2+aaaaaaaa")
+
+    assert Version.from_string("1.2.3+aaaaaaaa") == Version(1, 2, 3, 0xAAAAAAAA)
+    assert Version.from_string("10.2.3+12345678") == Version(10, 2, 3, 0x12345678)
+    assert Version.from_string("1.20.354+50") == Version(1, 20, 354, 0x50)
+
+    def round_trip(string: str):
+        assert str(Version.from_string(string)) == string
+
+    round_trip("1.2.3+aaaaaaaa")
+    round_trip("10.2.3+12345678")
+    round_trip("2.20.1+00000aaa")