diff --git a/frigate/models.py b/frigate/models.py index 0d5f1ab4a..4bd795026 100644 --- a/frigate/models.py +++ b/frigate/models.py @@ -66,3 +66,8 @@ class Recordings(Model): # type: ignore[misc] motion = IntegerField(null=True) objects = IntegerField(null=True) segment_size = FloatField(default=0) # this should be stored as MB + + +# Used for temporary table in record/cleanup.py +class RecordingsToDelete(Model): # type: ignore[misc] + id = CharField(null=False, primary_key=False, max_length=30) diff --git a/frigate/record/cleanup.py b/frigate/record/cleanup.py index 074494a7f..855d6462c 100644 --- a/frigate/record/cleanup.py +++ b/frigate/record/cleanup.py @@ -7,21 +7,17 @@ import os import threading from pathlib import Path -from peewee import chunked, DoesNotExist, DatabaseError, Model, CharField +from peewee import chunked, DoesNotExist, DatabaseError from multiprocessing.synchronize import Event as MpEvent from frigate.config import RetainModeEnum, FrigateConfig from frigate.const import RECORD_DIR, SECONDS_IN_DAY -from frigate.models import Event, Recordings, Timeline +from frigate.models import Event, Recordings, Timeline, RecordingsToDelete from frigate.record.util import remove_empty_directories logger = logging.getLogger(__name__) -class RecordingsToDelete(Model): - id = CharField(null=False, primary_key=False, max_length=30) - - class RecordingCleanup(threading.Thread): """Cleanup existing recordings based on retention config."""