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

Rename enum

Browse Source
This commit is contained in:
Nick Mowen 2023-04-30 08:40:44 -06:00
parent ca90946e2e
commit bbc2c7743d
3 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/events.py
View File

@ -22,7 +22,7 @@ from typing import Dict
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class DetectionTypeEnum(str, Enum): class EventTypeEnum(str, Enum):
# api = "api" # api = "api"
# audio = "audio" # audio = "audio"
tracked_object = "tracked_object" tracked_object = "tracked_object"
@ -91,7 +91,7 @@ class EventProcessor(threading.Thread):
) )
) )
if source_type == DetectionTypeEnum.tracked_object: if source_type == EventTypeEnum.tracked_object:
if event_type == "start": if event_type == "start":
self.events_in_process[event_data["id"]] = event_data self.events_in_process[event_data["id"]] = event_data
continue continue

8
frigate/object_processing.py
View File

@ -21,7 +21,7 @@ from frigate.config import (
FrigateConfig, FrigateConfig,
) )
from frigate.const import CLIPS_DIR from frigate.const import CLIPS_DIR
from frigate.events import DetectionTypeEnum from frigate.events import EventTypeEnum
from frigate.util import ( from frigate.util import (
SharedMemoryFrameManager, SharedMemoryFrameManager,
calculate_region, calculate_region,
@ -658,7 +658,7 @@ class TrackedObjectProcessor(threading.Thread):
def start(camera, obj: TrackedObject, current_frame_time): def start(camera, obj: TrackedObject, current_frame_time):
self.event_queue.put( self.event_queue.put(
(DetectionTypeEnum.tracked_object, "start", camera, obj.to_dict()) (EventTypeEnum.tracked_object, "start", camera, obj.to_dict())
) )
def update(camera, obj: TrackedObject, current_frame_time): def update(camera, obj: TrackedObject, current_frame_time):
@ -674,7 +674,7 @@ class TrackedObjectProcessor(threading.Thread):
obj.previous = after obj.previous = after
self.event_queue.put( self.event_queue.put(
( (
DetectionTypeEnum.tracked_object, EventTypeEnum.tracked_object,
"update", "update",
camera, camera,
obj.to_dict(include_thumbnail=True), obj.to_dict(include_thumbnail=True),
@ -732,7 +732,7 @@ class TrackedObjectProcessor(threading.Thread):
self.event_queue.put( self.event_queue.put(
( (
DetectionTypeEnum.tracked_object, EventTypeEnum.tracked_object,
"end", "end",
camera, camera,
obj.to_dict(include_thumbnail=True), obj.to_dict(include_thumbnail=True),

4
frigate/timeline.py
View File

@ -5,7 +5,7 @@ import threading
import queue import queue
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.events import DetectionTypeEnum from frigate.events import EventTypeEnum
from frigate.models import Timeline from frigate.models import Timeline
from multiprocessing.queues import Queue from multiprocessing.queues import Queue
@ -44,7 +44,7 @@ class TimelineProcessor(threading.Thread):
except queue.Empty: except queue.Empty:
continue continue
if input_type == DetectionTypeEnum.tracked_object: if input_type == EventTypeEnum.tracked_object:
self.handle_object_detection( self.handle_object_detection(
camera, event_type, prev_event_data, event_data camera, event_type, prev_event_data, event_data
) )