bakery-ia/shared/messaging/events.py

"""
Event definitions for microservices communication
- Simple class-based approach to avoid dataclass issues
"""

from datetime import datetime, timezone
from typing import Dict, Any, Optional
import uuid

class BaseEvent:
    """Base event class"""
    def __init__(self, service_name: str, data: Dict[str, Any], event_type: str = "", correlation_id: Optional[str] = None):
        self.service_name = service_name
        self.data = data
        self.event_type = event_type
        self.event_id = str(uuid.uuid4())
        self.timestamp = datetime.now(timezone.utc)
        self.correlation_id = correlation_id
    
    def to_dict(self) -> Dict[str, Any]: # Add this method
        """Converts the event object to a dictionary for JSON serialization."""
        return {
            "service_name": self.service_name,
            "data": self.data,
            "event_type": self.event_type,
            "event_id": self.event_id,
            "timestamp": self.timestamp.isoformat(), # Convert datetime to ISO 8601 string
            "correlation_id": self.correlation_id
        }

# Training Events
class TrainingStartedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="training.started",
            correlation_id=correlation_id
        )

class TrainingCompletedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="training.completed",
            correlation_id=correlation_id
        )

class TrainingFailedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="training.failed",
            correlation_id=correlation_id
        )

# Forecasting Events
class ForecastGeneratedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="forecast.generated",
            correlation_id=correlation_id
        )

class ForecastRequestedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="forecast.requested",
            correlation_id=correlation_id
        )

# User Events
class UserRegisteredEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="user.registered",
            correlation_id=correlation_id
        )

class UserLoginEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="user.login",
            correlation_id=correlation_id
        )

# Tenant Events
class TenantCreatedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="tenant.created",
            correlation_id=correlation_id
        )

class TenantUpdatedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="tenant.updated",
            correlation_id=correlation_id
        )

# Notification Events
class NotificationSentEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="notification.sent",
            correlation_id=correlation_id
        )

class NotificationFailedEvent(BaseEvent):
    def __init__(self, service_name: str, data: Dict[str, Any], correlation_id: Optional[str] = None):
        super().__init__(
            service_name=service_name,
            data=data,
            event_type="notification.failed",
            correlation_id=correlation_id
        )