python-terminusgps 1.5.9__tar.gz → 1.5.10__tar.gz

{python_terminusgps-1.5.9 → python_terminusgps-1.5.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-terminusgps
-Version: 1.5.9
+Version: 1.5.10
 Summary: Provides abstractions/utilities for working with Wialon API, Authorize.NET API, AWS API, and more.
 Project-URL: Documentation, https://app.terminusgps.com/docs/apps/python-terminusgps/index.html
 Project-URL: Repository, https://github.com/terminusgps/python-terminusgps

{python_terminusgps-1.5.9 → python_terminusgps-1.5.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-terminusgps"
-version = "1.5.9"
+version = "1.5.10"
 description = "Provides abstractions/utilities for working with Wialon API, Authorize.NET API, AWS API, and more."
 readme = "README.md"
 authors = [ {name = "Blake Nall", email = "blake@terminusgps.com" } ]

{python_terminusgps-1.5.9 → python_terminusgps-1.5.10}/terminusgps/wialon/utils.py

@@ -1,9 +1,33 @@
 import secrets
 import string
 
+from typing import Any
 from .session import WialonSession
 from .flags import DATAFLAG_UNIT_BASE
 
+from terminusgps.wialon.items import (
+    WialonUser,
+    WialonUnit,
+    WialonUnitGroup,
+    WialonResource,
+)
+
+
+def get_wialon_cls(items_type: str) -> Any:
+    """Returns a Wialon object class based on items_type."""
+    match items_type:
+        case "user":
+            wialon_cls = WialonUser
+        case "avl_unit":
+            wialon_cls = WialonUnit
+        case "avl_unit_group":
+            wialon_cls = WialonUnitGroup
+        case "avl_resource":
+            wialon_cls = WialonResource
+        case _:
+            raise ValueError(f"Invalid items_type '{items_type}'")
+    return wialon_cls
+
 
 def is_unique(value: str, session: WialonSession, items_type: str = "avl_unit") -> bool:
     """Determines if the value is unique among Wialon objects of type 'items_type'."""
@@ -50,9 +74,9 @@ def get_id_from_iccid(iccid: str, session: WialonSession) -> str | None:
             "to": 0,
         }
     )
-    if response.get("totalItemsCount", 0) != 1:
-        return None
-    return response["items"][0].get("id")
+
+    if response.get("totalItemsCount", 0) == 1:
+        return response["items"][0].get("id")
 
 
 def main() -> None: