util: argparse: device ID helper class

Add a helper class for specifying Infuse IoT device IDs that evaluates
to an integer but opaquely handles different ID forms.

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

Author: JordanYates

src/infuse_iot/util/argparse.py

@@ -79,3 +79,13 @@ def to_ctype(cls, addr_type: rpc_enum_bt_le_addr_type, value: int) -> rpc_struct
     def integer_value(cls, string) -> int:
         """Integer value from address string"""
         return cast(int, cls(string))
+
+
+class InfuseDeviceId:
+    """Infuse-IoT Device ID"""
+
+    def __new__(cls, string) -> int:  # type: ignore
+        try:
+            return int(string, 16)
+        except ValueError as e:
+            raise argparse.ArgumentTypeError(f"{string} is not a valid hex ID") from e

tests/util/test_argparse.py

@@ -6,7 +6,7 @@
 
 import pytest
 
-from infuse_iot.util.argparse import BtLeAddress, ValidDir, ValidFile
+from infuse_iot.util.argparse import BtLeAddress, InfuseDeviceId, ValidDir, ValidFile
 
 assert "TOXTEMPDIR" in os.environ, "you must run these tests using tox"
 
@@ -46,3 +46,16 @@ def test_bt_le_address():
     assert isinstance(addr, int)
     addr = BtLeAddress("123456aaFF4A")
     assert isinstance(addr, int)
+
+
+def test_infuse_device_id():
+    with pytest.raises(argparse.ArgumentTypeError):
+        InfuseDeviceId("NotHex")
+    with pytest.raises(argparse.ArgumentTypeError):
+        InfuseDeviceId("aabb::00")
+    assert InfuseDeviceId("0x00aa") == 0xAA
+    assert InfuseDeviceId("00aa") == 0xAA
+    assert InfuseDeviceId("0x99") == 0x99
+    assert InfuseDeviceId("99") == 0x99
+    assert InfuseDeviceId("0x1234aa43bc") == 0x1234AA43BC
+    assert InfuseDeviceId("1234aa43bc") == 0x1234AA43BC