feat(ansible): AnsibleTarget model + 0015_ansible_inventory_targets migration

steward/models/__init__.py

@@ -5,6 +5,7 @@ from .monitors import PingResult, DnsResult, PingStatus, DnsStatus
 from .metrics import PluginMetric
 from .alerts import AlertRule, AlertState, AlertEvent, AlertOperator, AlertStateEnum
 from .ansible import AnsibleRun, AnsibleRunStatus
+from .ansible_inventory import AnsibleTarget, AnsibleGroup, ansible_target_groups
 from .settings import AppSetting
 from .dashboard import Dashboard, DashboardWidget, DashboardShareToken
 
@@ -15,6 +16,7 @@ __all__ = [
     "PluginMetric",
     "AlertRule", "AlertState", "AlertEvent", "AlertOperator", "AlertStateEnum",
     "AnsibleRun", "AnsibleRunStatus",
+    "AnsibleTarget", "AnsibleGroup", "ansible_target_groups",
     "AppSetting",
     "Dashboard", "DashboardWidget", "DashboardShareToken",
 ]

steward/models/ansible_inventory.py

@@ -0,0 +1,69 @@
+from __future__ import annotations
+import uuid
+from datetime import datetime, timezone
+from sqlalchemy import Column, DateTime, ForeignKey, JSON, String, Table
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+from .base import Base
+
+ansible_target_groups = Table(
+    "ansible_target_groups",
+    Base.metadata,
+    Column(
+        "target_id",
+        String(36),
+        ForeignKey("ansible_targets.id", ondelete="CASCADE"),
+        primary_key=True,
+    ),
+    Column(
+        "group_id",
+        String(36),
+        ForeignKey("ansible_groups.id", ondelete="CASCADE"),
+        primary_key=True,
+    ),
+)
+
+
+class AnsibleTarget(Base):
+    __tablename__ = "ansible_targets"
+
+    id: Mapped[str] = mapped_column(
+        String(36), primary_key=True, default=lambda: str(uuid.uuid4())
+    )
+    name: Mapped[str] = mapped_column(String(128), nullable=False, unique=True)
+    address: Mapped[str] = mapped_column(String(255), nullable=False)
+    ansible_vars: Mapped[dict] = mapped_column(JSON, nullable=False, default=dict)
+    host_id: Mapped[str | None] = mapped_column(
+        String(36), ForeignKey("hosts.id", ondelete="SET NULL"), nullable=True
+    )
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True),
+        nullable=False,
+        default=lambda: datetime.now(timezone.utc),
+    )
+    groups: Mapped[list["AnsibleGroup"]] = relationship(
+        "AnsibleGroup",
+        secondary=ansible_target_groups,
+        back_populates="targets",
+        lazy="select",
+    )
+
+
+class AnsibleGroup(Base):
+    __tablename__ = "ansible_groups"
+
+    id: Mapped[str] = mapped_column(
+        String(36), primary_key=True, default=lambda: str(uuid.uuid4())
+    )
+    name: Mapped[str] = mapped_column(String(128), nullable=False, unique=True)
+    ansible_vars: Mapped[dict] = mapped_column(JSON, nullable=False, default=dict)
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True),
+        nullable=False,
+        default=lambda: datetime.now(timezone.utc),
+    )
+    targets: Mapped[list["AnsibleTarget"]] = relationship(
+        "AnsibleTarget",
+        secondary=ansible_target_groups,
+        back_populates="groups",
+        lazy="select",
+    )