fix validator tests

This commit is contained in:
YS 2022-01-20 15:29:44 +03:00
parent c742c2703a
commit 34f8018a27

View File

@ -408,8 +408,6 @@ class CameraGStreamerInput(CameraInput):
) )
class CameraInputValidator:
@validator("inputs")
def validate_roles(cls, v): def validate_roles(cls, v):
roles = [role for i in v for role in i.roles] roles = [role for i in v for role in i.roles]
roles_set = set(roles) roles_set = set(roles)
@ -423,13 +421,21 @@ class CameraInputValidator:
return v return v
class CameraFfmpegConfig(FfmpegConfig, CameraInputValidator): class CameraFfmpegConfig(FfmpegConfig):
inputs: List[CameraFFmpegInput] = Field(title="Camera FFMpeg inputs.") 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.") inputs: List[CameraGStreamerInput] = Field(title="Camera GStreamer inputs.")
@validator("inputs")
def validate_roles(cls, v):
return validate_roles(cls, v)
class SnapshotsConfig(FrigateBaseModel): class SnapshotsConfig(FrigateBaseModel):
enabled: bool = Field(default=False, title="Snapshots enabled.") enabled: bool = Field(default=False, title="Snapshots enabled.")