mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-30 19:09:13 +01:00
84a0827aee
Use config data classes to eliminate some of the boilerplate associated with setting up the configuration. In particular, using dataclasses removes a lot of the boilerplate around assigning properties to the object and allows these to be easily immutable by freezing them. In the case of simple, non-nested dataclasses, this also provides more convenient `asdict` helpers. To set this up, where previously the objects would be parsed from the config via the `__init__` method, create a `build` classmethod that does this and calls the dataclass initializer. Some of the objects are mutated at runtime, in particular some of the zones are mutated to set the color (this might be able to be refactored out) and some of the camera functionality can be enabled/disabled. Some of the configs with `enabled` properties don't seem to have mqtt hooks to be able to toggle this, in particular, the clips, snapshots, and detect can be toggled but rtmp and record configs do not, but all of these configs are still not frozen in case there is some other functionality I am missing. There are a couple other minor fixes here, one that was introduced by me recently where `max_seconds` was not defined, the other to properly `get()` the message payload when handling publishing mqtt messages sent via websocket.
342 lines
12 KiB
Python
342 lines
12 KiB
Python
import json
|
|
from unittest import TestCase, main
|
|
import voluptuous as vol
|
|
from frigate.config import FRIGATE_CONFIG_SCHEMA, FrigateConfig
|
|
|
|
|
|
class TestConfig(TestCase):
|
|
def setUp(self):
|
|
self.minimal = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
}
|
|
},
|
|
}
|
|
|
|
def test_empty(self):
|
|
FRIGATE_CONFIG_SCHEMA({})
|
|
|
|
def test_minimal(self):
|
|
FRIGATE_CONFIG_SCHEMA(self.minimal)
|
|
|
|
def test_config_class(self):
|
|
FrigateConfig(config=self.minimal)
|
|
|
|
def test_inherit_tracked_objects(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"objects": {"track": ["person", "dog"]},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "dog" in frigate_config.cameras["back"].objects.track
|
|
|
|
def test_override_tracked_objects(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"objects": {"track": ["person", "dog"]},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"objects": {"track": ["cat"]},
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "cat" in frigate_config.cameras["back"].objects.track
|
|
|
|
def test_default_object_filters(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"objects": {"track": ["person", "dog"]},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "dog" in frigate_config.cameras["back"].objects.filters
|
|
|
|
def test_inherit_object_filters(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"objects": {
|
|
"track": ["person", "dog"],
|
|
"filters": {"dog": {"threshold": 0.7}},
|
|
},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "dog" in frigate_config.cameras["back"].objects.filters
|
|
assert frigate_config.cameras["back"].objects.filters["dog"].threshold == 0.7
|
|
|
|
def test_override_object_filters(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"objects": {
|
|
"track": ["person", "dog"],
|
|
"filters": {"dog": {"threshold": 0.7}},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "dog" in frigate_config.cameras["back"].objects.filters
|
|
assert frigate_config.cameras["back"].objects.filters["dog"].threshold == 0.7
|
|
|
|
def test_global_object_mask(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"objects": {"track": ["person", "dog"]},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"objects": {
|
|
"mask": "0,0,1,1,0,1",
|
|
"filters": {"dog": {"mask": "1,1,1,1,1,1"}},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "dog" in frigate_config.cameras["back"].objects.filters
|
|
assert len(frigate_config.cameras["back"].objects.filters["dog"].raw_mask) == 2
|
|
assert (
|
|
len(frigate_config.cameras["back"].objects.filters["person"].raw_mask) == 1
|
|
)
|
|
|
|
def test_ffmpeg_params_global(self):
|
|
config = {
|
|
"ffmpeg": {"input_args": ["-re"]},
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"objects": {
|
|
"track": ["person", "dog"],
|
|
"filters": {"dog": {"threshold": 0.7}},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "-re" in frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"]
|
|
|
|
def test_ffmpeg_params_camera(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
],
|
|
"input_args": ["-re"],
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"objects": {
|
|
"track": ["person", "dog"],
|
|
"filters": {"dog": {"threshold": 0.7}},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "-re" in frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"]
|
|
|
|
def test_ffmpeg_params_input(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{
|
|
"path": "rtsp://10.0.0.1:554/video",
|
|
"roles": ["detect"],
|
|
"input_args": ["-re"],
|
|
}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"objects": {
|
|
"track": ["person", "dog"],
|
|
"filters": {"dog": {"threshold": 0.7}},
|
|
},
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert "-re" in frigate_config.cameras["back"].ffmpeg_cmds[0]["cmd"]
|
|
|
|
def test_inherit_clips_retention(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"clips": {"retain": {"default": 20, "objects": {"person": 30}}},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
}
|
|
},
|
|
}
|
|
frigate_config = FrigateConfig(config=config)
|
|
assert frigate_config.cameras["back"].clips.retain.objects["person"] == 30
|
|
|
|
def test_roles_listed_twice_throws_error(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"clips": {"retain": {"default": 20, "objects": {"person": 30}}},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]},
|
|
{"path": "rtsp://10.0.0.1:554/video2", "roles": ["detect"]},
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
}
|
|
},
|
|
}
|
|
self.assertRaises(vol.MultipleInvalid, lambda: FrigateConfig(config=config))
|
|
|
|
def test_zone_matching_camera_name_throws_error(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"clips": {"retain": {"default": 20, "objects": {"person": 30}}},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"zones": {"back": {"coordinates": "1,1,1,1,1,1"}},
|
|
}
|
|
},
|
|
}
|
|
self.assertRaises(vol.MultipleInvalid, lambda: FrigateConfig(config=config))
|
|
|
|
def test_clips_should_default_to_global_objects(self):
|
|
config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"clips": {"retain": {"default": 20, "objects": {"person": 30}}},
|
|
"objects": {"track": ["person", "dog"]},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
"clips": {"enabled": True},
|
|
}
|
|
},
|
|
}
|
|
config = FrigateConfig(config=config)
|
|
assert config.cameras["back"].clips.objects is None
|
|
|
|
def test_role_assigned_but_not_enabled(self):
|
|
json_config = {
|
|
"mqtt": {"host": "mqtt"},
|
|
"cameras": {
|
|
"back": {
|
|
"ffmpeg": {
|
|
"inputs": [
|
|
{
|
|
"path": "rtsp://10.0.0.1:554/video",
|
|
"roles": ["detect", "rtmp"],
|
|
},
|
|
{"path": "rtsp://10.0.0.1:554/record", "roles": ["record"]},
|
|
]
|
|
},
|
|
"height": 1080,
|
|
"width": 1920,
|
|
}
|
|
},
|
|
}
|
|
|
|
config = FrigateConfig(config=json_config)
|
|
ffmpeg_cmds = config.cameras["back"].ffmpeg_cmds
|
|
assert len(ffmpeg_cmds) == 1
|
|
assert not "clips" in ffmpeg_cmds[0]["roles"]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(verbosity=2)
|