uiprotect 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

uiprotect/api.py

@@ -65,6 +65,13 @@ from .utils import (
 )
 from .websocket import Websocket
 
+if sys.version_info[:2] < (3, 13):
+    from http import cookies
+
+    # See: https://github.com/python/cpython/issues/112713
+    cookies.Morsel._reserved["partitioned"] = "partitioned"  # type: ignore[attr-defined]
+    cookies.Morsel._flags.add("partitioned")  # type: ignore[attr-defined]
+
 TOKEN_COOKIE_MAX_EXP_SECONDS = 60
 
 NEVER_RAN = -1000

uiprotect/data/bootstrap.py

@@ -22,6 +22,7 @@ from ..utils import normalize_mac, utc_now
 from .base import (
     RECENT_EVENT_MAX,
     ProtectBaseObject,
+    ProtectDeviceModel,
     ProtectModel,
     ProtectModelWithId,
 )
@@ -65,6 +66,7 @@ STATS_KEYS = {
     "recordingSchedules",
 }
 IGNORE_DEVICE_KEYS = {"nvrMac", "guid"}
+STATS_AND_IGNORE_DEVICE_KEYS = STATS_KEYS | IGNORE_DEVICE_KEYS
 
 CAMERA_EVENT_ATTR_MAP: dict[EventType, tuple[str, str]] = {
     EventType.MOTION: ("last_motion", "last_motion_event_id"),
@@ -78,56 +80,36 @@ CAMERA_EVENT_ATTR_MAP: dict[EventType, tuple[str, str]] = {
 }
 
 
-def _remove_stats_keys(data: dict[str, Any]) -> None:
-    for key in STATS_KEYS.intersection(data):
-        del data[key]
-
-
 def _process_light_event(event: Event) -> None:
     if event.light is None:
         return
 
-    dt = event.light.last_motion
-    if dt is None or event.start >= dt or (event.end is not None and event.end >= dt):
+    if _event_is_in_range(event, event.light.last_motion):
         event.light.last_motion_event_id = event.id
 
 
+def _event_is_in_range(event: Event, dt: datetime | None) -> bool:
+    """Check if event is in range of datetime."""
+    return (
+        dt is None or event.start >= dt or (event.end is not None and event.end >= dt)
+    )
+
+
 def _process_sensor_event(event: Event) -> None:
     if event.sensor is None:
         return
-
-    if event.type == EventType.MOTION_SENSOR:
-        dt = event.sensor.motion_detected_at
-        if (
-            dt is None
-            or event.start >= dt
-            or (event.end is not None and event.end >= dt)
-        ):
+    if event.type is EventType.MOTION_SENSOR:
+        if _event_is_in_range(event, event.sensor.motion_detected_at):
             event.sensor.last_motion_event_id = event.id
     elif event.type in {EventType.SENSOR_CLOSED, EventType.SENSOR_OPENED}:
-        dt = event.sensor.open_status_changed_at
-        if (
-            dt is None
-            or event.start >= dt
-            or (event.end is not None and event.end >= dt)
-        ):
+        if _event_is_in_range(event, event.sensor.open_status_changed_at):
             event.sensor.last_contact_event_id = event.id
-    elif event.type == EventType.SENSOR_EXTREME_VALUE:
-        dt = event.sensor.extreme_value_detected_at
-        if (
-            dt is None
-            or event.start >= dt
-            or (event.end is not None and event.end >= dt)
-        ):
+    elif event.type is EventType.SENSOR_EXTREME_VALUE:
+        if _event_is_in_range(event, event.sensor.extreme_value_detected_at):
             event.sensor.extreme_value_detected_at = event.end
             event.sensor.last_value_event_id = event.id
-    elif event.type == EventType.SENSOR_ALARM:
-        dt = event.sensor.alarm_triggered_at
-        if (
-            dt is None
-            or event.start >= dt
-            or (event.end is not None and event.end >= dt)
-        ):
+    elif event.type is EventType.SENSOR_ALARM:
+        if _event_is_in_range(event, event.sensor.alarm_triggered_at):
             event.sensor.last_value_event_id = event.id
 
 
@@ -144,21 +126,25 @@ def _process_camera_event(event: Event) -> None:
 
     event_type = event.type
     dt_attr, event_attr = CAMERA_EVENT_ATTR_MAP[event_type]
-    dt = getattr(camera, dt_attr)
-    if dt is None or event.start >= dt or (event.end is not None and event.end >= dt):
-        setattr(camera, event_attr, event.id)
-        setattr(camera, dt_attr, event.start)
-        if event_type in _CAMERA_SMART_AND_LINE_EVENTS:
-            for smart_type in event.smart_detect_types:
-                camera.last_smart_detect_event_ids[smart_type] = event.id
-                camera.last_smart_detects[smart_type] = event.start
-        elif event_type is _CAMERA_SMART_AUDIO_EVENT:
-            for smart_type in event.smart_detect_types:
-                audio_type = smart_type.audio_type
-                if audio_type is None:
-                    continue
-                camera.last_smart_audio_detect_event_ids[audio_type] = event.id
-                camera.last_smart_audio_detects[audio_type] = event.start
+    dt: datetime | None = getattr(camera, dt_attr)
+    if not _event_is_in_range(event, dt):
+        return
+
+    event_id = event.id
+    event_start = event.start
+
+    setattr(camera, event_attr, event_id)
+    setattr(camera, dt_attr, event_start)
+    if event_type in _CAMERA_SMART_AND_LINE_EVENTS:
+        for smart_type in event.smart_detect_types:
+            camera.last_smart_detect_event_ids[smart_type] = event_id
+            camera.last_smart_detects[smart_type] = event_start
+    elif event_type is _CAMERA_SMART_AUDIO_EVENT:
+        for smart_type in event.smart_detect_types:
+            if (audio_type := smart_type.audio_type) is None:
+                continue
+            camera.last_smart_audio_detect_event_ids[audio_type] = event_id
+            camera.last_smart_audio_detects[audio_type] = event_start
 
 
 @dataclass
@@ -324,7 +310,7 @@ class Bootstrap(ProtectBaseObject):
         if ref is None:
             return None
 
-        devices = getattr(self, f"{ref.model.value}s")
+        devices: dict[str, ProtectModelWithId] = getattr(self, f"{ref.model.value}s")
         return cast(ProtectAdoptableDeviceModel, devices.get(ref.id))
 
     def get_device_from_id(self, device_id: str) -> ProtectAdoptableDeviceModel | None:
@@ -332,7 +318,7 @@ class Bootstrap(ProtectBaseObject):
         ref = self.id_lookup.get(device_id)
         if ref is None:
             return None
-        devices = getattr(self, f"{ref.model.value}s")
+        devices: dict[str, ProtectModelWithId] = getattr(self, f"{ref.model.value}s")
         return cast(ProtectAdoptableDeviceModel, devices.get(ref.id))
 
     def process_event(self, event: Event) -> None:
@@ -363,14 +349,6 @@ class Bootstrap(ProtectBaseObject):
                 ),
             )
 
-    def _get_frame_data(
-        self,
-        packet: WSPacket,
-    ) -> tuple[dict[str, Any], dict[str, Any] | None]:
-        if self.capture_ws_stats:
-            return deepcopy(packet.action_frame.data), deepcopy(packet.data_frame.data)
-        return packet.action_frame.data, packet.data_frame.data
-
     def _process_add_packet(
         self,
         packet: WSPacket,
@@ -412,7 +390,7 @@ class Bootstrap(ProtectBaseObject):
 
     def _process_remove_packet(self, packet: WSPacket) -> WSSubscriptionMessage | None:
         model: str | None = packet.action_frame.data.get("modelKey")
-        devices = getattr(self, f"{model}s", None)
+        devices: dict[str, ProtectDeviceModel] | None = getattr(self, f"{model}s", None)
 
         if devices is None:
             return None
@@ -439,7 +417,8 @@ class Bootstrap(ProtectBaseObject):
         ignore_stats: bool,
     ) -> WSSubscriptionMessage | None:
         if ignore_stats:
-            _remove_stats_keys(data)
+            for key in STATS_KEYS.intersection(data):
+                del data[key]
         # nothing left to process
         if not data:
             self._create_stat(packet, None, True)
@@ -477,9 +456,10 @@ class Bootstrap(ProtectBaseObject):
         ignore_stats: bool,
     ) -> WSSubscriptionMessage | None:
         model_type = action["modelKey"]
-        if ignore_stats:
-            _remove_stats_keys(data)
-        for key in IGNORE_DEVICE_KEYS.intersection(data):
+        remove_keys = (
+            STATS_AND_IGNORE_DEVICE_KEYS if ignore_stats else IGNORE_DEVICE_KEYS
+        )
+        for key in remove_keys.intersection(data):
             del data[key]
         # `last_motion` from cameras update every 100 milliseconds when a motion event is active
         # this overrides the behavior to only update `last_motion` when a new event starts
@@ -491,12 +471,12 @@ class Bootstrap(ProtectBaseObject):
             return None
 
         key = f"{model_type}s"
-        devices = getattr(self, key)
+        devices: dict[str, ProtectModelWithId] = getattr(self, key)
         action_id: str = action["id"]
         if action_id in devices:
             if action_id not in devices:
                 raise ValueError(f"Unknown device update for {model_type}: {action_id}")
-            obj: ProtectModelWithId = devices[action_id]
+            obj = devices[action_id]
             data = obj.unifi_dict_to_dict(data)
             old_obj = obj.copy()
             obj = obj.update_from_dict(deepcopy(data))
@@ -542,7 +522,12 @@ class Bootstrap(ProtectBaseObject):
         ignore_stats: bool = False,
     ) -> WSSubscriptionMessage | None:
         """Process a WS packet."""
-        action, data = self._get_frame_data(packet)
+        action = packet.action_frame.data
+        data = packet.data_frame.data
+        if self.capture_ws_stats:
+            action = deepcopy(action)
+            data = deepcopy(data)
+
         new_update_id: str = action["newUpdateId"]
         if new_update_id is not None:
             self.last_update_id = new_update_id
@@ -629,7 +614,9 @@ class Bootstrap(ProtectBaseObject):
         if isinstance(device, NVR):
             self.nvr = device
         else:
-            devices = getattr(self, f"{model_type.value}s")
+            devices: dict[str, ProtectModelWithId] = getattr(
+                self, f"{model_type.value}s"
+            )
             devices[device.id] = device
         _LOGGER.debug("Successfully refresh model: %s %s", model_type, device_id)
 

{uiprotect-1.0.0.dist-info → uiprotect-1.1.0.dist-info}/METADATA

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

{uiprotect-1.0.0.dist-info → uiprotect-1.1.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 uiprotect/__init__.py,sha256=llnQNtiBfwQG8IkQXovvFz4LZeFjrJx7XdmmUhu3a9E,289
 uiprotect/__main__.py,sha256=C_bHCOkv5qj6WMy-6ELoY3Y6HDhLxOa1a30CzmbZhsg,462
-uiprotect/api.py,sha256=EgqupTzTKsbSNck-h6vYpptibYtAHhFcxG6WviX5R8w,66173
+uiprotect/api.py,sha256=hN1PvNhOTFIfOmBtnUCfpujIXBEBYIKttaYnqlS_2es,66463
 uiprotect/cli/__init__.py,sha256=sSLW9keVQOkgFcMW18HTDjRrt9sJ0KWjn9DJDA6f9Pc,8658
 uiprotect/cli/backup.py,sha256=ZiS7RZnJGKI8TJKLW2cOUzkRM8nyTvE5Ov_jZZGtvSM,36708
 uiprotect/cli/base.py,sha256=zpTm2kyJe_GLixnv3Uadke__iRLh64AEwQzp-2hqS7g,7730
@@ -15,7 +15,7 @@ uiprotect/cli/sensors.py,sha256=fQtcDJCVxs4VbAqcavgBy2ABiVxAW3GXtna6_XFBp2k,8153
 uiprotect/cli/viewers.py,sha256=2cyrp104ffIvgT0wYGIO0G35QMkEbFe7fSVqLwDXQYQ,2171
 uiprotect/data/__init__.py,sha256=OcfuJl2qXfHcj_mdnrHhzZ5tEIZrw8auziX5IE7dn-I,2938
 uiprotect/data/base.py,sha256=p75RcPjVTmHEk6BVWApougBC2RBwUdFq8x_gL4zr8xc,37313
-uiprotect/data/bootstrap.py,sha256=lxijgYGOEhGzbgnBaJdmCJcmzEKIJnGbIBx_LQzNEMU,21547
+uiprotect/data/bootstrap.py,sha256=NH4jSEm44M0aijmViLHip2PKhVchuZJEeqWI3bcZwq0,21285
 uiprotect/data/convert.py,sha256=rOQplUMIdTMD2SbAx_iI9BNPDscnhDvyRVLEMDhtADg,2047
 uiprotect/data/devices.py,sha256=Nq3bOko5PFf5LvEBoD4JV8kmbq50laRdh3VHMWX7t-0,111809
 uiprotect/data/nvr.py,sha256=aIi4d0KCJ55r1EP-33sBhvEa8lx-41jDGrDpAq5VnOE,47598
@@ -30,8 +30,8 @@ uiprotect/test_util/__init__.py,sha256=d2g7afa0LSdixQ0kjEDYwafDFME_UlW2LzxpamZ2B
 uiprotect/test_util/anonymize.py,sha256=f-8ijU-_y9r-uAbhIPn0f0I6hzJpAkvJzc8UpWihObI,8478
 uiprotect/utils.py,sha256=6OLY8hNiCzk418PjJJIlFW7jjPzVt1vxBKEzBSqMeTk,18418
 uiprotect/websocket.py,sha256=IzDPyqbzrkOMREvahN-e2zdvVD0VABSCWy6jSoCwOT0,7299
-uiprotect-1.0.0.dist-info/LICENSE,sha256=INx18jhdbVXMEiiBANeKEbrbz57ckgzxk5uutmmcxGk,1111
-uiprotect-1.0.0.dist-info/METADATA,sha256=TkJEgMl3hpJViK8WveWGO2O2OvPbUihSdiCEecfOcWc,10984
-uiprotect-1.0.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-uiprotect-1.0.0.dist-info/entry_points.txt,sha256=J78AUTPrTTxgI3s7SVgrmGqDP7piX2wuuEORzhDdVRA,47
-uiprotect-1.0.0.dist-info/RECORD,,
+uiprotect-1.1.0.dist-info/LICENSE,sha256=INx18jhdbVXMEiiBANeKEbrbz57ckgzxk5uutmmcxGk,1111
+uiprotect-1.1.0.dist-info/METADATA,sha256=--s4fFpQ0KbuDx6VYuPnkko1w4_f2hRpGR-IJOvZrt4,10984
+uiprotect-1.1.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+uiprotect-1.1.0.dist-info/entry_points.txt,sha256=J78AUTPrTTxgI3s7SVgrmGqDP7piX2wuuEORzhDdVRA,47
+uiprotect-1.1.0.dist-info/RECORD,,