diff --git a/frigate/config/camera/record.py b/frigate/config/camera/record.py index d984b6387..2be076f8e 100644 --- a/frigate/config/camera/record.py +++ b/frigate/config/camera/record.py @@ -32,6 +32,7 @@ class RecordRetainConfig(FrigateBaseModel): days: float = Field(default=0, title="Default retention period.") mode: RetainModeEnum = Field(default=RetainModeEnum.all, title="Retain mode.") + class RecordCleanupConfig(FrigateBaseModel): trigger_minutes: int = Field( default=60, diff --git a/frigate/test/test_storage.py b/frigate/test/test_storage.py index 55f5873c6..bdec99ca9 100644 --- a/frigate/test/test_storage.py +++ b/frigate/test/test_storage.py @@ -140,13 +140,15 @@ class TestHttp(unittest.TestCase): assert storage.camera_storage_stats == { "front_door": {"bandwidth": 0, "needs_refresh": True}, } - - @patch('frigate.storage.shutil.disk_usage') + + @patch("frigate.storage.shutil.disk_usage") def test_storage_cleanup(self, mock_disk_usage): """Ensure that all recordings are cleaned up when necessary.""" config = FrigateConfig(**self.minimal_config) storage = StorageMaintainer(config, MagicMock()) - mock_disk_usage.return_value = shutil._ntuple_diskusage(total=10000, used=5000, free=5000) + mock_disk_usage.return_value = shutil._ntuple_diskusage( + total=10000, used=5000, free=5000 + ) id = "123456.keep" time_keep = datetime.datetime.now().timestamp() @@ -213,12 +215,14 @@ class TestHttp(unittest.TestCase): Recordings.get(Recordings.id == rec_d2_id) Recordings.get(Recordings.id == rec_d3_id) - @patch('frigate.storage.shutil.disk_usage') + @patch("frigate.storage.shutil.disk_usage") def test_storage_cleanup_keeps_retained(self, mock_disk_usage): """Ensure that all recordings are cleaned up when necessary.""" config = FrigateConfig(**self.minimal_config) storage = StorageMaintainer(config, MagicMock()) - mock_disk_usage.return_value = shutil._ntuple_diskusage(total=10000, used=5000, free=5000) + mock_disk_usage.return_value = shutil._ntuple_diskusage( + total=10000, used=5000, free=5000 + ) id = "123456.keep" time_keep = datetime.datetime.now().timestamp()