mirror of
https://github.com/DRYTRIX/TimeTracker.git
synced 2026-05-19 04:40:32 -05:00
Dries Peeters
b4486a627f
- Webhook models: remove duplicate index definitions so db.create_all() no longer raises 'index already exists' (columns already have index=True) - ImportService: fix circular import by late-importing ClientService, ProjectService, TimeTrackingService in __init__ - reports: fix F823 by renaming unpack variable _ to _entry_count to avoid shadowing gettext _ in export_task_excel() - Code quality: add .flake8 with extend-ignore so flake8 CI passes; simplify pyproject.toml isort config (drop unsupported options) - Format: run black and isort on app/ - tests: restore minimal app fixture in test_import_export_models
129 lines
3.6 KiB
Python
129 lines
3.6 KiB
Python
"""
|
|
Event bus for domain events.
|
|
Provides decoupled event-driven architecture.
|
|
"""
|
|
|
|
from functools import wraps
|
|
from typing import Any, Callable, Dict, List
|
|
|
|
from flask import current_app
|
|
|
|
from app.constants import WebhookEvent
|
|
|
|
|
|
class EventBus:
|
|
"""Simple event bus for domain events"""
|
|
|
|
def __init__(self):
|
|
self._handlers: Dict[str, List[Callable]] = {}
|
|
|
|
def subscribe(self, event_type: str, handler: Callable) -> None:
|
|
"""
|
|
Subscribe a handler to an event type.
|
|
|
|
Args:
|
|
event_type: Event type (e.g., 'time_entry.created')
|
|
handler: Function to call when event is emitted
|
|
"""
|
|
if event_type not in self._handlers:
|
|
self._handlers[event_type] = []
|
|
self._handlers[event_type].append(handler)
|
|
|
|
def unsubscribe(self, event_type: str, handler: Callable) -> None:
|
|
"""Unsubscribe a handler from an event type"""
|
|
if event_type in self._handlers:
|
|
try:
|
|
self._handlers[event_type].remove(handler)
|
|
except ValueError:
|
|
pass
|
|
|
|
def emit(self, event_type: str, data: Dict[str, Any]) -> None:
|
|
"""
|
|
Emit an event to all subscribed handlers.
|
|
|
|
Args:
|
|
event_type: Event type
|
|
data: Event data
|
|
"""
|
|
handlers = self._handlers.get(event_type, [])
|
|
for handler in handlers:
|
|
try:
|
|
handler(event_type, data)
|
|
except Exception as e:
|
|
current_app.logger.error(f"Error in event handler for {event_type}: {e}", exc_info=True)
|
|
|
|
def clear(self) -> None:
|
|
"""Clear all event handlers"""
|
|
self._handlers.clear()
|
|
|
|
|
|
# Global event bus instance
|
|
_event_bus = EventBus()
|
|
|
|
|
|
def get_event_bus() -> EventBus:
|
|
"""Get the global event bus instance"""
|
|
return _event_bus
|
|
|
|
|
|
def emit_event(event_type: str, data: Dict[str, Any]) -> None:
|
|
"""
|
|
Emit an event using the global event bus.
|
|
|
|
Args:
|
|
event_type: Event type
|
|
data: Event data
|
|
"""
|
|
_event_bus.emit(event_type, data)
|
|
|
|
|
|
def subscribe_to_event(event_type: str):
|
|
"""
|
|
Decorator to subscribe a function to an event type.
|
|
|
|
Usage:
|
|
@subscribe_to_event('time_entry.created')
|
|
def handle_time_entry_created(event_type, data):
|
|
# Handle event
|
|
"""
|
|
|
|
def decorator(func: Callable) -> Callable:
|
|
_event_bus.subscribe(event_type, func)
|
|
return func
|
|
|
|
return decorator
|
|
|
|
|
|
# Example event handlers
|
|
@subscribe_to_event(WebhookEvent.TIME_ENTRY_CREATED.value)
|
|
def handle_time_entry_created(event_type: str, data: Dict[str, Any]) -> None:
|
|
"""Handle time entry created event"""
|
|
try:
|
|
from app.utils.webhook_dispatcher import dispatch_webhook
|
|
|
|
dispatch_webhook(event_type, data)
|
|
except Exception as e:
|
|
current_app.logger.error(f"Failed to dispatch webhook for {event_type}: {e}")
|
|
|
|
|
|
@subscribe_to_event(WebhookEvent.PROJECT_CREATED.value)
|
|
def handle_project_created(event_type: str, data: Dict[str, Any]) -> None:
|
|
"""Handle project created event"""
|
|
try:
|
|
from app.utils.webhook_dispatcher import dispatch_webhook
|
|
|
|
dispatch_webhook(event_type, data)
|
|
except Exception as e:
|
|
current_app.logger.error(f"Failed to dispatch webhook for {event_type}: {e}")
|
|
|
|
|
|
@subscribe_to_event(WebhookEvent.INVOICE_CREATED.value)
|
|
def handle_invoice_created(event_type: str, data: Dict[str, Any]) -> None:
|
|
"""Handle invoice created event"""
|
|
try:
|
|
from app.utils.webhook_dispatcher import dispatch_webhook
|
|
|
|
dispatch_webhook(event_type, data)
|
|
except Exception as e:
|
|
current_app.logger.error(f"Failed to dispatch webhook for {event_type}: {e}")
|