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

Set sub label on object data if event is in progress

Browse Source
This commit is contained in:
Nick Mowen 2022-09-23 07:05:52 -06:00
parent 3846a13805
commit 938482a6da
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/http.py
View File

@ -28,6 +28,7 @@ from playhouse.shortcuts import model_to_dict
from frigate.const import CLIPS_DIR from frigate.const import CLIPS_DIR
from frigate.models import Event, Recordings from frigate.models import Event, Recordings
from frigate.object_processing import TrackedObjectProcessor
from frigate.stats import stats_snapshot from frigate.stats import stats_snapshot
from frigate.version import VERSION from frigate.version import VERSION
@ -211,7 +212,7 @@ def delete_retain(id):
@bp.route("/events/<id>/sub_label", methods=("POST",)) @bp.route("/events/<id>/sub_label", methods=("POST",))
def set_sub_label(id): def set_sub_label(id):
try: try:
event = Event.get(Event.id == id) event: Event = Event.get(Event.id == id)
except DoesNotExist: except DoesNotExist:
return make_response( return make_response(
jsonify({"success": False, "message": "Event " + id + " not found"}), 404 jsonify({"success": False, "message": "Event " + id + " not found"}), 404
@ -234,6 +235,9 @@ def set_sub_label(id):
400, 400,
) )
if not event.end_time:
current_app.detected_frames_processor.camera_states[event.camera].tracked_objects[event.id].obj_data["sub_label"] = new_sub_label
event.sub_label = new_sub_label event.sub_label = new_sub_label
event.save() event.save()
return make_response( return make_response(