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

convert to upsert

Browse Source
This commit is contained in:
Blake Blackshear 2023-04-24 06:27:41 -05:00
parent df5d18f1ef
commit 48ebdde8e3
1 changed files with 27 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
frigate/events.py
View File

@ -135,24 +135,33 @@ class EventProcessor(threading.Thread):
if self.events_in_process[event_data["id"]]["has_snapshot"]: if self.events_in_process[event_data["id"]]["has_snapshot"]:
event_data["has_snapshot"] = True event_data["has_snapshot"] = True
Event.replace( event = {
id=event_data["id"], Event.id: event_data["id"],
label=event_data["label"], Event.label: event_data["label"],
camera=camera, Event.camera: camera,
start_time=start_time, Event.start_time: start_time,
end_time=end_time, Event.end_time: end_time,
top_score=event_data["top_score"], Event.top_score: event_data["top_score"],
score=score, Event.score: score,
zones=list(event_data["entered_zones"]), Event.zones: list(event_data["entered_zones"]),
thumbnail=event_data["thumbnail"], Event.thumbnail: event_data["thumbnail"],
region=region, Event.region: region,
box=box, Event.box: box,
has_clip=event_data["has_clip"], Event.has_clip: event_data["has_clip"],
has_snapshot=event_data["has_snapshot"], Event.has_snapshot: event_data["has_snapshot"],
model_hash=first_detector.model.model_hash, Event.model_hash: first_detector.model.model_hash,
model_type=first_detector.model.model_type, Event.model_type: first_detector.model.model_type,
detector_type=first_detector.type, Event.detector_type: first_detector.type,
).execute() }
(
Event.insert(event)
.on_conflict(
conflict_target=[Event.id],
update=event,
)
.execute()
)
# update the stored copy for comparison on future update messages # update the stored copy for comparison on future update messages
self.events_in_process[event_data["id"]] = event_data self.events_in_process[event_data["id"]] = event_data