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 ) workstream_id: Mapped[uuid.UUID | None] = mapped_column( UUID(as_uuid=True), ForeignKey("workstreams.id", ondelete="SET NULL"), nullable=True ) domain: Mapped["Domain"] = relationship("Domain", lazy="selectin") # noqa: F821 topic: Mapped["Topic"] = relationship("Topic", lazy="selectin") # noqa: F821 workstream: Mapped["Workstream"] = relationship("Workstream", lazy="selectin") # noqa: F821 @property def domain_slug(self) -> str: return self.domain.slug if self.domain is not None else ""