state-hub/api/models/extension_point.py

import enum
import uuid

from sqlalchemy import Enum, ForeignKey, String, Text
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import Mapped, mapped_column, relationship

from api.models.base import Base, TimestampMixin, new_uuid


class EPStatus(str, enum.Enum):
    open = "open"
    in_progress = "in_progress"
    addressed = "addressed"
    deferred = "deferred"
    wont_fix = "wont_fix"


class ExtensionPoint(Base, TimestampMixin):
    __tablename__ = "extension_points"

    id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), primary_key=True, default=new_uuid
    )
    ep_id: Mapped[str | None] = mapped_column(
        String(30), nullable=True, unique=True, index=True
    )  # human-readable ref, e.g. EP-CUST-001
    domain_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True),
        ForeignKey("domains.id", ondelete="RESTRICT"),
        nullable=False,
        index=True,
    )
    title: Mapped[str] = mapped_column(String(255), nullable=False)
    description: Mapped[str | None] = mapped_column(Text, nullable=True)
    location: Mapped[str | None] = mapped_column(String(500), nullable=True)
    ep_type: Mapped[str] = mapped_column(
        String(50), nullable=False, default="other"
    )  # api | schema | mcp | dashboard | architecture | integration | other
    status: Mapped[EPStatus] = mapped_column(
        Enum(EPStatus, name="epstatus"), nullable=False, default=EPStatus.open
    )
    priority: Mapped[str] = mapped_column(String(20), nullable=False, default="medium")
    topic_id: Mapped[uuid.UUID | None] = mapped_column(
        UUID(as_uuid=True), ForeignKey("topics.id", ondelete="SET NULL"), nullable=True
    )
    workplan_id: Mapped[uuid.UUID | None] = mapped_column(
        UUID(as_uuid=True), ForeignKey("workplans.id", ondelete="SET NULL"), nullable=True
    )

    domain: Mapped["Domain"] = relationship("Domain", lazy="selectin")  # noqa: F821
    topic: Mapped["Topic"] = relationship("Topic", lazy="selectin")  # noqa: F821
    workplan: Mapped["Workplan"] = relationship("Workplan", lazy="selectin")  # noqa: F821

    @property
    def domain_slug(self) -> str:
        return self.domain.slug if self.domain is not None else ""