mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-21 19:07:46 +01:00
bfa7a5cc60
* Store camera labels in dict and other optimizations * Add max on timeout so it is at least 60 * Ensure db timeout is at least 60 * Update list once a day to ensure new labels are cleaned up * Formatting * Insert recordings as bulk instead of individually. * Fix * Refactor event and timeline cleanup * Remove unused
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""Run recording maintainer and cleanup."""
|
|
|
|
import logging
|
|
import multiprocessing as mp
|
|
import signal
|
|
import threading
|
|
from types import FrameType
|
|
from typing import Optional
|
|
|
|
from playhouse.sqliteq import SqliteQueueDatabase
|
|
from setproctitle import setproctitle
|
|
|
|
from frigate.config import FrigateConfig
|
|
from frigate.models import Event, Recordings, RecordingsToDelete, Timeline
|
|
from frigate.record.cleanup import RecordingCleanup
|
|
from frigate.record.maintainer import RecordingMaintainer
|
|
from frigate.types import FeatureMetricsTypes
|
|
from frigate.util.services import listen
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def manage_recordings(
|
|
config: FrigateConfig,
|
|
object_recordings_info_queue: mp.Queue,
|
|
audio_recordings_info_queue: mp.Queue,
|
|
process_info: dict[str, FeatureMetricsTypes],
|
|
) -> None:
|
|
stop_event = mp.Event()
|
|
|
|
def receiveSignal(signalNumber: int, frame: Optional[FrameType]) -> None:
|
|
stop_event.set()
|
|
|
|
signal.signal(signal.SIGTERM, receiveSignal)
|
|
signal.signal(signal.SIGINT, receiveSignal)
|
|
|
|
threading.current_thread().name = "process:recording_manager"
|
|
setproctitle("frigate.recording_manager")
|
|
listen()
|
|
|
|
db = SqliteQueueDatabase(
|
|
config.database.path,
|
|
pragmas={
|
|
"auto_vacuum": "FULL", # Does not defragment database
|
|
"cache_size": -512 * 1000, # 512MB of cache
|
|
"synchronous": "NORMAL", # Safe when using WAL https://www.sqlite.org/pragma.html#pragma_synchronous
|
|
},
|
|
timeout=max(60, 10 * len([c for c in config.cameras.values() if c.enabled])),
|
|
)
|
|
models = [Event, Recordings, Timeline, RecordingsToDelete]
|
|
db.bind(models)
|
|
|
|
maintainer = RecordingMaintainer(
|
|
config,
|
|
object_recordings_info_queue,
|
|
audio_recordings_info_queue,
|
|
process_info,
|
|
stop_event,
|
|
)
|
|
maintainer.start()
|
|
|
|
cleanup = RecordingCleanup(config, stop_event)
|
|
cleanup.start()
|