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

Support setting sub label scores via API

Browse Source
This commit is contained in:
Nicolas Mowen 2023-07-10 07:38:29 -06:00 committed by Nick Mowen
parent 7c0d25f9da
commit 5ba3fc9fb5
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/http.py
View File

@ -370,10 +370,9 @@ def set_sub_label(id):
jsonify({"success": False, "message": "Event " + id + " not found"}), 404 jsonify({"success": False, "message": "Event " + id + " not found"}), 404
) )
if request.json: json: dict[str, any] = request.get_json(silent=True) or {}
new_sub_label = request.json.get("subLabel") new_sub_label = json.get("subLabel")
else: new_score = json.get("subLabelScore")
new_sub_label = None
if new_sub_label and len(new_sub_label) > 100: if new_sub_label and len(new_sub_label) > 100:
return make_response( return make_response(
@ -395,9 +394,15 @@ def set_sub_label(id):
) )
if tracked_obj: if tracked_obj:
tracked_obj.obj_data["sub_label"] = new_sub_label tracked_obj.obj_data["sub_label"] = (new_sub_label, new_score)
event.sub_label = new_sub_label event.sub_label = new_sub_label
if new_score:
data = event.data
data["sub_label_score"] = new_score
event.data = data
event.save() event.save()
return make_response( return make_response(
jsonify( jsonify(