administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-02 01:05:20 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix validator tests

Browse Source
This commit is contained in:
YS 2022-01-20 15:29:44 +03:00
parent c742c2703a
commit 34f8018a27
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
frigate/config.py
View File

@ -408,28 +408,34 @@ class CameraGStreamerInput(CameraInput):
)
class CameraInputValidator:
@validator("inputs")
def validate_roles(cls, v):
roles = [role for i in v for role in i.roles]
roles_set = set(roles)
def validate_roles(cls, v):
roles = [role for i in v for role in i.roles]
roles_set = set(roles)
if len(roles) > len(roles_set):
raise ValueError("Each input role may only be used once.")
if len(roles) > len(roles_set):
raise ValueError("Each input role may only be used once.")
if not "detect" in roles:
raise ValueError("The detect role is required.")
if not "detect" in roles:
raise ValueError("The detect role is required.")
return v
return v
class CameraFfmpegConfig(FfmpegConfig, CameraInputValidator):
class CameraFfmpegConfig(FfmpegConfig):
inputs: List[CameraFFmpegInput] = Field(title="Camera FFMpeg inputs.")
@validator("inputs")
def validate_roles(cls, v):
return validate_roles(cls, v)
class CameraGStreamerConfig(GstreamerConfig, CameraInputValidator):
class CameraGStreamerConfig(GstreamerConfig):
inputs: List[CameraGStreamerInput] = Field(title="Camera GStreamer inputs.")
@validator("inputs")
def validate_roles(cls, v):
return validate_roles(cls, v)
class SnapshotsConfig(FrigateBaseModel):
enabled: bool = Field(default=False, title="Snapshots enabled.")