administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-19 09:27:06 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix formatting

Browse Source
This commit is contained in:
Jake 2024-12-13 11:19:52 -05:00
parent cf27811323
commit 52d2c309f5
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
frigate/config/camera/record.py
View File

@ -32,6 +32,7 @@ class RecordRetainConfig(FrigateBaseModel):
days: float = Field(default=0, title="Default retention period.") days: float = Field(default=0, title="Default retention period.")
mode: RetainModeEnum = Field(default=RetainModeEnum.all, title="Retain mode.") mode: RetainModeEnum = Field(default=RetainModeEnum.all, title="Retain mode.")
class RecordCleanupConfig(FrigateBaseModel): class RecordCleanupConfig(FrigateBaseModel):
trigger_minutes: int = Field( trigger_minutes: int = Field(
default=60, default=60,

14
frigate/test/test_storage.py
View File

@ -140,13 +140,15 @@ class TestHttp(unittest.TestCase):
assert storage.camera_storage_stats == { assert storage.camera_storage_stats == {
"front_door": {"bandwidth": 0, "needs_refresh": True}, "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): def test_storage_cleanup(self, mock_disk_usage):
"""Ensure that all recordings are cleaned up when necessary.""" """Ensure that all recordings are cleaned up when necessary."""
config = FrigateConfig(**self.minimal_config) config = FrigateConfig(**self.minimal_config)
storage = StorageMaintainer(config, MagicMock()) 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" id = "123456.keep"
time_keep = datetime.datetime.now().timestamp() 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_d2_id)
Recordings.get(Recordings.id == rec_d3_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): def test_storage_cleanup_keeps_retained(self, mock_disk_usage):
"""Ensure that all recordings are cleaned up when necessary.""" """Ensure that all recordings are cleaned up when necessary."""
config = FrigateConfig(**self.minimal_config) config = FrigateConfig(**self.minimal_config)
storage = StorageMaintainer(config, MagicMock()) 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" id = "123456.keep"
time_keep = datetime.datetime.now().timestamp() time_keep = datetime.datetime.now().timestamp()