__init__.py

from app.db.docs.admin_settings import (
    AuditLogDocument,
    SystemSettingsDocument,
)
from app.db.docs.dlq import DLQMessageDocument
from app.db.docs.event import (
    EventArchiveDocument,
    EventDocument,
)
from app.db.docs.execution import ExecutionDocument, ResourceUsage
from app.db.docs.notification import (
    NotificationDocument,
    NotificationSubscriptionDocument,
)
from app.db.docs.replay import (
    ReplayConfig,
    ReplaySessionDocument,
)
from app.db.docs.resource import ResourceAllocationDocument
from app.db.docs.saga import SagaDocument
from app.db.docs.saved_script import SavedScriptDocument
from app.db.docs.user import UserDocument
from app.db.docs.user_settings import (
    UserSettingsDocument,
    UserSettingsSnapshotDocument,
)

# All document classes that need to be initialized with Beanie
ALL_DOCUMENTS = [
    UserDocument,
    ExecutionDocument,
    SavedScriptDocument,
    NotificationDocument,
    NotificationSubscriptionDocument,
    UserSettingsDocument,
    UserSettingsSnapshotDocument,
    SagaDocument,
    DLQMessageDocument,
    EventDocument,
    EventArchiveDocument,
    ReplaySessionDocument,
    ResourceAllocationDocument,
    SystemSettingsDocument,
    AuditLogDocument,
]

__all__ = [
    # User
    "UserDocument",
    # Execution
    "ExecutionDocument",
    "ResourceUsage",
    # Saved Script
    "SavedScriptDocument",
    # Notification
    "NotificationDocument",
    "NotificationSubscriptionDocument",
    # User Settings
    "UserSettingsDocument",
    "UserSettingsSnapshotDocument",
    # Saga
    "SagaDocument",
    # DLQ
    "DLQMessageDocument",
    # Event
    "EventDocument",
    "EventArchiveDocument",
    # Replay
    "ReplaySessionDocument",
    "ReplayConfig",
    # Resource
    "ResourceAllocationDocument",
    # Admin Settings
    "SystemSettingsDocument",
    "AuditLogDocument",
    # All documents list for Beanie init
    "ALL_DOCUMENTS",
]