Browse Source

add change type to events topic

#476
Blake Blackshear 4 năm trước cách đây
mục cha
commit
d1d833ea9a
1 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 2 2
      frigate/object_processing.py

+ 2 - 2
frigate/object_processing.py

@@ -430,7 +430,7 @@ class TrackedObjectProcessor(threading.Thread):
 
         def update(camera, obj: TrackedObject, current_frame_time):
             after = obj.to_dict()
-            message = { 'before': obj.previous, 'after': after }
+            message = { 'before': obj.previous, 'after': after, 'type': 'new' if obj.previous['false_positive'] else 'update' }
             self.client.publish(f"{self.topic_prefix}/events", json.dumps(message), retain=False)
             obj.previous = after
 
@@ -439,7 +439,7 @@ class TrackedObjectProcessor(threading.Thread):
             event_data = obj.to_dict(include_thumbnail=True)
             event_data['has_snapshot'] = False
             if not obj.false_positive:
-                message = { 'before': obj.previous, 'after': obj.to_dict() }
+                message = { 'before': obj.previous, 'after': obj.to_dict(), 'type': 'end' }
                 self.client.publish(f"{self.topic_prefix}/events", json.dumps(message), retain=False)
                 # write snapshot to disk if enabled
                 if snapshot_config.enabled: