mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-11-01 18:53:59 -04:00
23c039b42d
* refactored event dispatching added EventDocumentType and EventOperation to Event added event listeners to bulk recipe changes overhauled shopping list item events to be more useful modified shopping list item repo to return more information * added internal documentation for event types * renamed message_types.py to event_types.py * added unique event id and fixed instantiation * generalized event listeners and publishers moved apprise publisher to new apprise event listener fixed duplicate message bug with apprise publisher * added JWT field for user-specified integration id * removed obselete test notification route * tuned up existing notification tests * added dependency to get integration_id from jwt * added base crud controller to facilitate events * simplified event publishing * temporarily fixed test notification
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
from typing import Optional
|
|
|
|
from fastapi import BackgroundTasks, Depends
|
|
from pydantic import UUID4
|
|
|
|
from mealie.core.config import get_app_settings
|
|
from mealie.db.db_setup import generate_session
|
|
from mealie.services.event_bus_service.event_bus_listeners import AppriseEventListener, EventListenerBase
|
|
|
|
from .event_types import Event, EventBusMessage, EventDocumentDataBase, EventTypes
|
|
|
|
settings = get_app_settings()
|
|
ALGORITHM = "HS256"
|
|
|
|
|
|
class EventSource:
|
|
event_type: str
|
|
item_type: str
|
|
item_id: UUID4 | int
|
|
kwargs: dict
|
|
|
|
def __init__(self, event_type: str, item_type: str, item_id: UUID4 | int, **kwargs) -> None:
|
|
self.event_type = event_type
|
|
self.item_type = item_type
|
|
self.item_id = item_id
|
|
self.kwargs = kwargs
|
|
|
|
def dict(self) -> dict:
|
|
return {
|
|
"event_type": self.event_type,
|
|
"item_type": self.item_type,
|
|
"item_id": str(self.item_id),
|
|
**self.kwargs,
|
|
}
|
|
|
|
|
|
class EventBusService:
|
|
def __init__(self, bg: BackgroundTasks, session=Depends(generate_session)) -> None:
|
|
self.bg = bg
|
|
self.session = session
|
|
self.group_id: UUID4 | None = None
|
|
|
|
self.listeners: list[EventListenerBase] = [AppriseEventListener(self.session, self.group_id)]
|
|
|
|
def dispatch(
|
|
self,
|
|
integration_id: str,
|
|
group_id: UUID4,
|
|
event_type: EventTypes,
|
|
document_data: Optional[EventDocumentDataBase],
|
|
message: str = "",
|
|
) -> None:
|
|
self.group_id = group_id
|
|
|
|
event = Event(
|
|
message=EventBusMessage.from_type(event_type, body=message),
|
|
event_type=event_type,
|
|
integration_id=integration_id,
|
|
document_data=document_data,
|
|
)
|
|
|
|
self.bg.add_task(self.publish_event, event=event)
|
|
|
|
def publish_event(self, event: Event) -> None:
|
|
"""Publishes the event to all listeners"""
|
|
for listener in self.listeners:
|
|
if subscribers := listener.get_subscribers(event):
|
|
listener.publish_to_subscribers(event, subscribers)
|