Milestone 3

backend/app/models/__init__.py

@@ -0,0 +1,73 @@
+"""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",
+]