Parcourir la source

add change type to events topic

#476
Blake Blackshear il y a 4 ans
Parent
commit
d1d833ea9a
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  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: