From b46132e6f847da2fe38c34d63d84f920f2163d63 Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Sat, 29 Apr 2023 12:48:46 -0600 Subject: [PATCH] Add source type to event queue --- frigate/object_processing.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/frigate/object_processing.py b/frigate/object_processing.py index db2745309..766054818 100644 --- a/frigate/object_processing.py +++ b/frigate/object_processing.py @@ -21,6 +21,7 @@ from frigate.config import ( FrigateConfig, ) from frigate.const import CLIPS_DIR +from frigate.timeline import TimelineSourceEnum from frigate.util import ( SharedMemoryFrameManager, calculate_region, @@ -656,7 +657,7 @@ class TrackedObjectProcessor(threading.Thread): self.last_motion_detected: dict[str, float] = {} def start(camera, obj: TrackedObject, current_frame_time): - self.event_queue.put(("start", camera, obj.to_dict())) + self.event_queue.put((TimelineSourceEnum.tracked_object, "start", camera, obj.to_dict())) def update(camera, obj: TrackedObject, current_frame_time): obj.has_snapshot = self.should_save_snapshot(camera, obj) @@ -670,7 +671,7 @@ class TrackedObjectProcessor(threading.Thread): self.dispatcher.publish("events", json.dumps(message), retain=False) obj.previous = after self.event_queue.put( - ("update", camera, obj.to_dict(include_thumbnail=True)) + (TimelineSourceEnum.tracked_object, "update", camera, obj.to_dict(include_thumbnail=True)) ) def end(camera, obj: TrackedObject, current_frame_time): @@ -722,7 +723,7 @@ class TrackedObjectProcessor(threading.Thread): } self.dispatcher.publish("events", json.dumps(message), retain=False) - self.event_queue.put(("end", camera, obj.to_dict(include_thumbnail=True))) + self.event_queue.put((TimelineSourceEnum.tracked_object, "end", camera, obj.to_dict(include_thumbnail=True))) def snapshot(camera, obj: TrackedObject, current_frame_time): mqtt_config: MqttConfig = self.config.cameras[camera].mqtt