uiprotect 1.8.0__tar.gz → 1.9.0__tar.gz

{uiprotect-1.8.0 → uiprotect-1.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: uiprotect
-Version: 1.8.0
+Version: 1.9.0
 Summary: Python API for Unifi Protect (Unofficial)
 Home-page: https://github.com/uilibs/uiprotect
 License: MIT

{uiprotect-1.8.0 → uiprotect-1.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "uiprotect"
-version = "1.8.0"
+version = "1.9.0"
 description = "Python API for Unifi Protect (Unofficial)"
 authors = ["UI Protect Maintainers <ui@koston.org>"]
 license = "MIT"

{uiprotect-1.8.0 → uiprotect-1.9.0}/src/uiprotect/data/bootstrap.py

@@ -193,8 +193,8 @@ class Bootstrap(ProtectBaseObject):
         )
         data["macLookup"] = {}
         data["idLookup"] = {}
-        for model_type in ModelType.bootstrap_models_types_set():
-            key = model_type.devices_key
+        for model_type in ModelType.bootstrap_models_types_set:
+            key = model_type.devices_key  # type: ignore[attr-defined]
             items: dict[str, ProtectModel] = {}
             for item in data[key]:
                 if (
@@ -230,8 +230,8 @@ class Bootstrap(ProtectBaseObject):
         if "idLookup" in data:
             del data["idLookup"]
 
-        for model_type in ModelType.bootstrap_models_types_set():
-            attr = model_type.devices_key
+        for model_type in ModelType.bootstrap_models_types_set:
+            attr = model_type.devices_key  # type: ignore[attr-defined]
             if attr in data and isinstance(data[attr], dict):
                 data[attr] = list(data[attr].values())
 
@@ -353,7 +353,7 @@ class Bootstrap(ProtectBaseObject):
             if TYPE_CHECKING:
                 assert isinstance(obj, NVR)
             self.nvr = obj
-        elif model_type in ModelType.bootstrap_models_types_set():
+        elif model_type in ModelType.bootstrap_models_types_set:
             if TYPE_CHECKING:
                 assert isinstance(obj, ProtectAdoptableDeviceModel)
             if not self._api.ignore_unadopted or (
@@ -550,7 +550,7 @@ class Bootstrap(ProtectBaseObject):
             if action_action == "update":
                 if model_type is ModelType.NVR:
                     return self._process_nvr_update(packet, data, ignore_stats)
-                if model_type in ModelType.bootstrap_models_types_and_event_set():
+                if model_type in ModelType.bootstrap_models_types_and_event_set:
                     return self._process_device_update(
                         model_type,
                         packet,

{uiprotect-1.8.0 → uiprotect-1.9.0}/src/uiprotect/data/types.py

@@ -109,6 +109,12 @@ class ModelType(str, UnknownValuesEnumMixin, enum.Enum):
     RECORDING_SCHEDULE = "recordingSchedule"
     UNKNOWN = "unknown"
 
+    bootstrap_model_types: tuple[ModelType, ...]
+    bootstrap_models: tuple[str, ...]
+    bootstrap_models_set: set[str]
+    bootstrap_models_types_set: set[ModelType]
+    bootstrap_models_types_and_event_set: set[ModelType]
+
     @cached_property
     def devices_key(self) -> str:
         """Return the devices key."""
@@ -119,9 +125,8 @@ class ModelType(str, UnknownValuesEnumMixin, enum.Enum):
     def from_string(cls, value: str) -> ModelType:
         return cls(value)
 
-    @staticmethod
-    @cache
-    def bootstrap_model_types() -> tuple[ModelType, ...]:
+    @classmethod
+    def _bootstrap_model_types(cls) -> tuple[ModelType, ...]:
         """Return the bootstrap models as a tuple."""
         # TODO:
         # legacyUFV
@@ -139,31 +144,40 @@ class ModelType(str, UnknownValuesEnumMixin, enum.Enum):
             ModelType.CHIME,
         )
 
-    @staticmethod
-    @cache
-    def bootstrap_models() -> tuple[str, ...]:
+    @classmethod
+    def _bootstrap_models(cls) -> tuple[str, ...]:
         """Return the bootstrap models strings as a tuple."""
         return tuple(
-            model_type.value for model_type in ModelType.bootstrap_model_types()
+            model_type.value for model_type in ModelType._bootstrap_model_types()
         )
 
-    @staticmethod
-    @cache
-    def bootstrap_models_set() -> set[str]:
+    @classmethod
+    def _bootstrap_models_set(cls) -> set[str]:
         """Return the set of bootstrap models strings as a set."""
-        return set(ModelType.bootstrap_models())
+        return set(ModelType._bootstrap_models())
 
-    @staticmethod
-    @cache
-    def bootstrap_models_types_set() -> set[ModelType]:
+    @classmethod
+    def _bootstrap_models_types_set(cls) -> set[ModelType]:
         """Return the set of bootstrap models as a set."""
-        return set(ModelType.bootstrap_model_types())
+        return set(ModelType._bootstrap_model_types())
 
-    @staticmethod
-    @cache
-    def bootstrap_models_types_and_event_set() -> set[ModelType]:
+    @classmethod
+    def _bootstrap_models_types_and_event_set(cls) -> set[ModelType]:
         """Return the set of bootstrap models and the event model as a set."""
-        return ModelType.bootstrap_models_types_set() | {ModelType.EVENT}
+        return ModelType._bootstrap_models_types_set() | {ModelType.EVENT}
+
+    def _immutable(self, name: str, value: Any) -> None:
+        raise AttributeError("Cannot modify ModelType")
+
+
+ModelType.bootstrap_model_types = ModelType._bootstrap_model_types()
+ModelType.bootstrap_models = ModelType._bootstrap_models()
+ModelType.bootstrap_models_set = ModelType._bootstrap_models_set()
+ModelType.bootstrap_models_types_set = ModelType._bootstrap_models_types_set()
+ModelType.bootstrap_models_types_and_event_set = (
+    ModelType._bootstrap_models_types_and_event_set()
+)
+ModelType.__setattr__ = ModelType._immutable  # type: ignore[method-assign, assignment]
 
 
 @enum.unique