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

Added responses for more review.py endpoints

Browse Source
This commit is contained in:
Rui Alves 2024-10-16 13:53:57 +01:00
parent a53f128d96
commit c42dc19524
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/api/defs/generic_response.py Normal file
View File

@ -0,0 +1,6 @@
from pydantic import BaseModel
class GenericResponse(BaseModel):
success: bool
message: str

7
frigate/api/review.py
View File

@ -13,6 +13,7 @@ from fastapi.responses import JSONResponse
from peewee import Case, DoesNotExist, fn, operator 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.review_query_parameters import ( from frigate.api.defs.review_query_parameters import (
ReviewActivityMotionQueryParams, ReviewActivityMotionQueryParams,
ReviewQueryParams, ReviewQueryParams,
@ -572,7 +573,7 @@ def audio_activity(params: ReviewActivityMotionQueryParams = Depends()):
return JSONResponse(content=normalized) return JSONResponse(content=normalized)
@router.get("/review/event/{event_id}") @router.get("/review/event/{event_id}", response_model=ReviewSegmentResponse)
def get_review_from_event(event_id: str): def get_review_from_event(event_id: str):
try: try:
return model_to_dict( return model_to_dict(
@ -584,7 +585,7 @@ def get_review_from_event(event_id: str):
return "Review item not found", 404 return "Review item not found", 404
@router.get("/review/{event_id}") @router.get("/review/{event_id}", response_model=ReviewSegmentResponse)
def get_review(event_id: str): def get_review(event_id: str):
try: try:
return model_to_dict(ReviewSegment.get(ReviewSegment.id == event_id)) return model_to_dict(ReviewSegment.get(ReviewSegment.id == event_id))
@ -592,7 +593,7 @@ def get_review(event_id: str):
return "Review item not found", 404 return "Review item not found", 404
@router.delete("/review/{event_id}/viewed") @router.delete("/review/{event_id}/viewed", response_model=GenericResponse)
def set_not_reviewed(event_id: str): def set_not_reviewed(event_id: str):
try: try:
review: ReviewSegment = ReviewSegment.get(ReviewSegment.id == event_id) review: ReviewSegment = ReviewSegment.get(ReviewSegment.id == event_id)