Metamorph/backend/app/models/__init__.py

"""ORM models — every module must be imported here so Alembic's autogenerate
can see them via `Base.metadata`.
"""

from app.models.auth import (
    Group,
    GroupPermission,
    Invitation,
    InvitationGroup,
    Permission,
    RefreshToken,
    User,
    UserGroup,
)
from app.models.evidence import EvidenceFile
from app.models.mission import (
    Mission,
    MissionCategory,
    MissionMember,
    MissionScenario,
    MissionTest,
    MissionTestMitreTag,
)
from app.models.mitre import (
    MitreSubtechnique,
    MitreTactic,
    MitreTechnique,
    MitreTechniqueTactic,
)
from app.models.notification import Notification
from app.models.setting import DetectionLevel, Setting
from app.models.template import (
    ScenarioTemplate,
    ScenarioTemplateTest,
    TestTemplate,
    TestTemplateMitreTag,
)

__all__ = [
    # auth
    "Group",
    "GroupPermission",
    "Invitation",
    "InvitationGroup",
    "Permission",
    "RefreshToken",
    "User",
    "UserGroup",
    # evidence
    "EvidenceFile",
    # mission
    "Mission",
    "MissionCategory",
    "MissionMember",
    "MissionScenario",
    "MissionTest",
    "MissionTestMitreTag",
    # mitre
    "MitreSubtechnique",
    "MitreTactic",
    "MitreTechnique",
    "MitreTechniqueTactic",
    # notification
    "Notification",
    # setting
    "DetectionLevel",
    "Setting",
    # template
    "ScenarioTemplate",
    "ScenarioTemplateTest",
    "TestTemplate",
    "TestTemplateMitreTag",
]