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

Adding support for SMART_DETECT_LINE events

Browse Source
This commit is contained in:
jimmydoh
2024-09-13 10:08:00 +10:00
committed by Sebastian Goscik
parent 80ad55d0d0
commit 1bb8496b30
4 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unifi_protect_backup/cli.py
View File

@@ -10,7 +10,7 @@ from unifi_protect_backup import __version__
from unifi_protect_backup.unifi_protect_backup_core import UnifiProtectBackup
from unifi_protect_backup.utils import human_readable_to_float
DETECTION_TYPES = ["motion", "person", "vehicle", "ring"]
DETECTION_TYPES = ["motion", "person", "vehicle", "ring", "line"]
def _parse_detection_types(ctx, param, value):

20
unifi_protect_backup/event_listener.py
View File

@@ -42,7 +42,9 @@ 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:
@@ -66,14 +68,24 @@ class EventListener:
EventType.MOTION,
EventType.SMART_DETECT,
EventType.RING,
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
):
logger.extra_debug(f"Skipping unwanted line event: {msg.new_obj.id}") # type: ignore
return
elif msg.new_obj.type is EventType.SMART_DETECT:
for event_smart_detection_type in msg.new_obj.smart_detect_types:
if event_smart_detection_type not in self.detection_types:
@@ -95,7 +107,9 @@ 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.

4
unifi_protect_backup/missing_event_checker.py
View File

@@ -65,7 +65,7 @@ class MissingEventChecker:
events_chunk = await self._protect.get_events(
start=start_time,
end=end_time,
types=[EventType.MOTION, EventType.SMART_DETECT, EventType.RING],
types=[EventType.MOTION, EventType.SMART_DETECT, EventType.RING, EventType.SMART_DETECT_LINE],
limit=chunk_size,
)
@@ -107,6 +107,8 @@ class MissingEventChecker:
return False
if event.type is EventType.RING and "ring" not in self.detection_types:
return False
if event.type is EventType.SMART_DETECT_LINE and "line" not in self.detection_types:
return False
elif event.type is EventType.SMART_DETECT:
for event_smart_detection_type in event.smart_detect_types:
if event_smart_detection_type not in self.detection_types:

3
unifi_protect_backup/unifi_protect_backup_core.py
View File

@@ -197,7 +197,7 @@ class UnifiProtectBackup:
# Start the uiprotect connection by calling `update`
logger.info("Connecting to Unifi Protect...")
delay = 5 # Start with a 5 second delay
delay = 5 # Start with a 5 second delay
max_delay = 3600 # 1 hour in seconds
for attempts in range(20):
@@ -214,7 +214,6 @@ class UnifiProtectBackup:
else:
raise ConnectionError("Failed to connect to UniFi Protect after 20 attempts")
# Add a lock to the protect client that can be used to prevent code accessing the client when it has
# lost connection
self._protect.connect_event = asyncio.Event()