mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-27 09:07:41 +03:00
71e18ae888
* Add enum for type of classification for objects * Update recognized license plate topic to be used as attribute updater * Update attribute for attribute type object classification * Cleanup
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
"""Facilitates communication between processes."""
|
|
|
|
import logging
|
|
from enum import Enum
|
|
from typing import Any
|
|
|
|
from .zmq_proxy import Publisher, Subscriber
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class EventMetadataTypeEnum(str, Enum):
|
|
all = ""
|
|
manual_event_create = "manual_event_create"
|
|
manual_event_end = "manual_event_end"
|
|
regenerate_description = "regenerate_description"
|
|
sub_label = "sub_label"
|
|
attribute = "attribute"
|
|
lpr_event_create = "lpr_event_create"
|
|
save_lpr_snapshot = "save_lpr_snapshot"
|
|
|
|
|
|
class EventMetadataPublisher(Publisher):
|
|
"""Simplifies receiving event metadata."""
|
|
|
|
topic_base = "event_metadata/"
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
|
|
def publish(self, topic: EventMetadataTypeEnum, payload: Any) -> None:
|
|
super().publish(payload, topic.value)
|
|
|
|
|
|
class EventMetadataSubscriber(Subscriber):
|
|
"""Simplifies receiving event metadata."""
|
|
|
|
topic_base = "event_metadata/"
|
|
|
|
def __init__(self, topic: EventMetadataTypeEnum) -> None:
|
|
super().__init__(topic.value)
|
|
|
|
def _return_object(self, topic: str, payload: tuple) -> tuple:
|
|
if payload is None:
|
|
return (None, None)
|
|
|
|
topic = EventMetadataTypeEnum[topic[len(self.topic_base) :]]
|
|
return (topic, payload)
|