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

Combine 2 models into generic

Browse Source
This commit is contained in:
Rui Alves 2024-10-21 16:06:04 +01:00
parent 24dbe64f48
commit 20af8994f5
2 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/api/defs/review_body.py
View File

@ -1,11 +1,6 @@
from pydantic import BaseModel, conlist, constr from pydantic import BaseModel, conlist, constr
class ReviewSetMultipleReviewedBody(BaseModel): class ReviewModifyMultipleBody(BaseModel):
# List of string with at least one element and each element with at least one char
ids: conlist(constr(min_length=1), min_length=1)
class ReviewDeleteMultipleReviewsBody(BaseModel):
# List of string with at least one element and each element with at least one char # List of string with at least one element and each element with at least one char
ids: conlist(constr(min_length=1), min_length=1) ids: conlist(constr(min_length=1), min_length=1)

9
frigate/api/review.py
View File

@ -13,10 +13,7 @@ from peewee import Case, DoesNotExist, fn, operator
from playhouse.shortcuts import model_to_dict from playhouse.shortcuts import model_to_dict
from frigate.api.defs.generic_response import GenericResponse from frigate.api.defs.generic_response import GenericResponse
from frigate.api.defs.review_body import ( from frigate.api.defs.review_body import ReviewModifyMultipleBody
ReviewDeleteMultipleReviewsBody,
ReviewSetMultipleReviewedBody,
)
from frigate.api.defs.review_query_parameters import ( from frigate.api.defs.review_query_parameters import (
ReviewActivityMotionQueryParams, ReviewActivityMotionQueryParams,
ReviewQueryParams, ReviewQueryParams,
@ -311,7 +308,7 @@ def review_summary(params: ReviewSummaryQueryParams = Depends()):
@router.post("/reviews/viewed", response_model=GenericResponse) @router.post("/reviews/viewed", response_model=GenericResponse)
def set_multiple_reviewed(body: ReviewSetMultipleReviewedBody): def set_multiple_reviewed(body: ReviewModifyMultipleBody):
ReviewSegment.update(has_been_reviewed=True).where( ReviewSegment.update(has_been_reviewed=True).where(
ReviewSegment.id << body.ids ReviewSegment.id << body.ids
).execute() ).execute()
@ -323,7 +320,7 @@ def set_multiple_reviewed(body: ReviewSetMultipleReviewedBody):
@router.post("/reviews/delete", response_model=GenericResponse) @router.post("/reviews/delete", response_model=GenericResponse)
def delete_reviews(body: ReviewDeleteMultipleReviewsBody): def delete_reviews(body: ReviewModifyMultipleBody):
list_of_ids = body.ids list_of_ids = body.ids
reviews = ( reviews = (
ReviewSegment.select( ReviewSegment.select(