jgaunt/unifi-protect-backup
1
0
Fork 0
mirror of https://github.com/ep1cman/unifi-protect-backup.git synced 2025-12-10 13:23:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

ruff formatting

Browse Source
This commit is contained in:
Sebastian Goscik
2024-10-26 22:11:42 +01:00
parent 78a2c3034d
commit 2c43149c99
2 changed files with 10 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
unifi_protect_backup/event_listener.py
View File

@@ -42,9 +42,7 @@ class EventListener:
async def start(self): async def start(self):
"""Main Loop.""" """Main Loop."""
logger.debug("Subscribed to websocket") logger.debug("Subscribed to websocket")
self._unsub_websocket_state = self._protect.subscribe_websocket_state( self._unsub_websocket_state = self._protect.subscribe_websocket_state(self._websocket_state_callback)
self._websocket_state_callback
)
self._unsub = self._protect.subscribe_websocket(self._websocket_callback) self._unsub = self._protect.subscribe_websocket(self._websocket_callback)
def _websocket_callback(self, msg: WSSubscriptionMessage) -> None: def _websocket_callback(self, msg: WSSubscriptionMessage) -> None:
@@ -71,19 +69,13 @@ class EventListener:
EventType.SMART_DETECT_LINE, EventType.SMART_DETECT_LINE,
]: ]:
return return
if ( if msg.new_obj.type is EventType.MOTION and "motion" not in self.detection_types:
msg.new_obj.type is EventType.MOTION
and "motion" not in self.detection_types
):
logger.extra_debug(f"Skipping unwanted motion detection event: {msg.new_obj.id}") # type: ignore logger.extra_debug(f"Skipping unwanted motion detection event: {msg.new_obj.id}") # type: ignore
return return
if msg.new_obj.type is EventType.RING and "ring" not in self.detection_types: if msg.new_obj.type is EventType.RING and "ring" not in self.detection_types:
logger.extra_debug(f"Skipping unwanted ring event: {msg.new_obj.id}") # type: ignore logger.extra_debug(f"Skipping unwanted ring event: {msg.new_obj.id}") # type: ignore
return return
if ( if msg.new_obj.type is EventType.SMART_DETECT_LINE and "line" not in self.detection_types:
msg.new_obj.type is EventType.SMART_DETECT_LINE
and "line" not in self.detection_types
):
logger.extra_debug(f"Skipping unwanted line event: {msg.new_obj.id}") # type: ignore logger.extra_debug(f"Skipping unwanted line event: {msg.new_obj.id}") # type: ignore
return return
elif msg.new_obj.type is EventType.SMART_DETECT: elif msg.new_obj.type is EventType.SMART_DETECT:
@@ -107,9 +99,7 @@ class EventListener:
if "-" in msg.new_obj.id: if "-" in msg.new_obj.id:
msg.new_obj.id = msg.new_obj.id.split("-")[0] msg.new_obj.id = msg.new_obj.id.split("-")[0]
logger.debug( logger.debug(f"Adding event {msg.new_obj.id} to queue (Current download queue={self._event_queue.qsize()})")
f"Adding event {msg.new_obj.id} to queue (Current download queue={self._event_queue.qsize()})"
)
def _websocket_state_callback(self, state: WebsocketState) -> None: def _websocket_state_callback(self, state: WebsocketState) -> None:
"""Callback for websocket state messages. """Callback for websocket state messages.

7
unifi_protect_backup/missing_event_checker.py
View File

@@ -65,7 +65,12 @@ class MissingEventChecker:
events_chunk = await self._protect.get_events( events_chunk = await self._protect.get_events(
start=start_time, start=start_time,
end=end_time, end=end_time,
types=[EventType.MOTION, EventType.SMART_DETECT, EventType.RING, EventType.SMART_DETECT_LINE], types=[
EventType.MOTION,
EventType.SMART_DETECT,
EventType.RING,
EventType.SMART_DETECT_LINE,
],
limit=chunk_size, limit=chunk_size,
) )