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(