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

Lookup camera names in pyunifiprotect instead of a local map

Browse Source 
Pyunifiprotect should keep the known cameras upto date, unlike the map
we were keeping before which was never updated after start.
This commit is contained in:
Sebastian Goscik
2022-02-21 00:03:00 +00:00
parent f81d57735f
commit e491965e04
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
unifi_protect_backup/unifi_protect_backup.py
View File

@@ -3,7 +3,7 @@ import asyncio
import logging import logging
import pathlib import pathlib
import shutil import shutil
from typing import Callable, Dict, Optional from typing import Callable, Optional
import aiocron import aiocron
from pyunifiprotect import ProtectApiClient from pyunifiprotect import ProtectApiClient
@@ -170,7 +170,6 @@ class UnifiProtectBackup:
verbose (int): How verbose to setup logging, see :func:`setup_logging` for details. verbose (int): How verbose to setup logging, see :func:`setup_logging` for details.
_download_queue (asyncio.Queue): Queue of events that need to be backed up _download_queue (asyncio.Queue): Queue of events that need to be backed up
_unsub (Callable): Unsubscribe from the websocket callback _unsub (Callable): Unsubscribe from the websocket callback
_camera_names (Dict[str, str]): A map of camera IDs -> camera names
""" """
def __init__( def __init__(
@@ -215,7 +214,6 @@ class UnifiProtectBackup:
) )
self._download_queue: asyncio.Queue = asyncio.Queue() self._download_queue: asyncio.Queue = asyncio.Queue()
self._unsub: Callable[[], None] self._unsub: Callable[[], None]
self._camera_names: Dict[str, str]
async def start(self): async def start(self):
"""Bootstrap the backup process and kick off the main loop. """Bootstrap the backup process and kick off the main loop.
@@ -232,8 +230,6 @@ class UnifiProtectBackup:
# Start the pyunifiprotect connection by calling `update` # Start the pyunifiprotect connection by calling `update`
logger.info("Connecting to Unifi Protect...") logger.info("Connecting to Unifi Protect...")
await self._protect.update() await self._protect.update()
# Get a mapping of camera ids -> names
self._camera_names = {camera.id: camera.name for camera in self._protect.bootstrap.cameras.values()}
# Subscribe to the websocket # Subscribe to the websocket
self._unsub = self._protect.subscribe_websocket(self._websocket_callback) self._unsub = self._protect.subscribe_websocket(self._websocket_callback)
@@ -337,7 +333,7 @@ class UnifiProtectBackup:
logger.info(f"Backing up event: {event.id}") logger.info(f"Backing up event: {event.id}")
logger.debug(f"Remaining Queue: {self._download_queue.qsize()}") logger.debug(f"Remaining Queue: {self._download_queue.qsize()}")
logger.debug(f" Camera: {self._camera_names[event.camera_id]}") logger.debug(f" Camera: {self._protect.bootstrap.cameras[event.camera_id].name}")
logger.debug(f" Type: {event.type}") logger.debug(f" Type: {event.type}")
logger.debug(f" Start: {event.start.strftime('%Y-%m-%dT%H-%M-%S')}") logger.debug(f" Start: {event.start.strftime('%Y-%m-%dT%H-%M-%S')}")
logger.debug(f" End: {event.end.strftime('%Y-%m-%dT%H-%M-%S')}") logger.debug(f" End: {event.end.strftime('%Y-%m-%dT%H-%M-%S')}")
@@ -427,7 +423,7 @@ class UnifiProtectBackup:
""" """
path = pathlib.Path(self.rclone_destination) path = pathlib.Path(self.rclone_destination)
assert isinstance(event.camera_id, str) assert isinstance(event.camera_id, str)
path /= self._camera_names[event.camera_id] # directory per camera path /= self._protect.bootstrap.cameras[event.camera_id].name # directory per camera
path /= event.start.strftime("%Y-%m-%d") # Directory per day path /= event.start.strftime("%Y-%m-%d") # Directory per day
file_name = f"{event.start.strftime('%Y-%m-%dT%H-%M-%S')} {event.type}" file_name = f"{event.start.strftime('%Y-%m-%dT%H-%M-%S')} {event.type}"