From 938482a6da9d3ad835feb8b5eb598684701e9d02 Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Fri, 23 Sep 2022 07:05:52 -0600 Subject: [PATCH] Set sub label on object data if event is in progress --- frigate/http.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/frigate/http.py b/frigate/http.py index f6c9ca003..c3348f68a 100644 --- a/frigate/http.py +++ b/frigate/http.py @@ -28,6 +28,7 @@ from playhouse.shortcuts import model_to_dict from frigate.const import CLIPS_DIR from frigate.models import Event, Recordings +from frigate.object_processing import TrackedObjectProcessor from frigate.stats import stats_snapshot from frigate.version import VERSION @@ -211,7 +212,7 @@ def delete_retain(id): @bp.route("/events//sub_label", methods=("POST",)) def set_sub_label(id): try: - event = Event.get(Event.id == id) + event: Event = Event.get(Event.id == id) except DoesNotExist: return make_response( jsonify({"success": False, "message": "Event " + id + " not found"}), 404 @@ -234,6 +235,9 @@ def set_sub_label(id): 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.save() return make_response(