bakery-ia/services/alert_processor/app/models/__init__.py

"""
Alert Processor Service Models Package

Import all models to ensure they are registered with SQLAlchemy Base.
"""

# Import AuditLog model for this service
from shared.security import create_audit_log_model
from shared.database.base import Base

# Create audit log model for this service
AuditLog = create_audit_log_model(Base)

# Import all models to register them with the Base metadata
from .events import (
    Alert,
    Notification,
    Recommendation,
    EventInteraction,
    AlertStatus,
    PriorityLevel,
    AlertTypeClass,
    NotificationType,
    RecommendationType,
)

# List all models for easier access
__all__ = [
    # New event models
    "Alert",
    "Notification",
    "Recommendation",
    "EventInteraction",
    # Enums
    "AlertStatus",
    "PriorityLevel",
    "AlertTypeClass",
    "NotificationType",
    "RecommendationType",
    # System
    "AuditLog",
]