jgaunt/unifi-protect-backup
1
0
Fork 0
mirror of https://github.com/ep1cman/unifi-protect-backup.git synced 2025-12-05 23:53:30 +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):
"""Main Loop."""
logger.debug("Subscribed to websocket")
self._unsub_websocket_state = self._protect.subscribe_websocket_state(
self._websocket_state_callback
)
self._unsub_websocket_state = self._protect.subscribe_websocket_state(self._websocket_state_callback)
self._unsub = self._protect.subscribe_websocket(self._websocket_callback)
def _websocket_callback(self, msg: WSSubscriptionMessage) -> None:
@@ -71,19 +69,13 @@ class EventListener:
EventType.SMART_DETECT_LINE,
]:
return
if (
msg.new_obj.type is EventType.MOTION
and "motion" not in self.detection_types
):
if 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
return
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
return
if (
msg.new_obj.type is EventType.SMART_DETECT_LINE
and "line" not in self.detection_types
):
if 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
return
elif msg.new_obj.type is EventType.SMART_DETECT:
@@ -107,9 +99,7 @@ class EventListener:
if "-" in msg.new_obj.id:
msg.new_obj.id = msg.new_obj.id.split("-")[0]
logger.debug(
f"Adding event {msg.new_obj.id} to queue (Current download queue={self._event_queue.qsize()})"
)
logger.debug(f"Adding event {msg.new_obj.id} to queue (Current download queue={self._event_queue.qsize()})")
def _websocket_state_callback(self, state: WebsocketState) -> None:
"""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(
start=start_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,
)