48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
|
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 TDStatus(str, enum.Enum):
|
||
|
|
open = "open"
|
||
|
|
in_progress = "in_progress"
|
||
|
|
resolved = "resolved"
|
||
|
|
deferred = "deferred"
|
||
|
|
wont_fix = "wont_fix"
|
||
|
|
|
||
|
|
|
||
|
|
class TechnicalDebt(Base, TimestampMixin):
|
||
|
|
__tablename__ = "technical_debt"
|
||
|
|
|
||
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
||
|
|
UUID(as_uuid=True), primary_key=True, default=new_uuid
|
||
|
|
)
|
||
|
|
td_id: Mapped[str | None] = mapped_column(
|
||
|
|
String(30), nullable=True, unique=True, index=True
|
||
|
|
) # human-readable ref, e.g. TD-CUST-001
|
||
|
|
domain: Mapped[str] = mapped_column(String(50), 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)
|
||
|
|
debt_type: Mapped[str] = mapped_column(
|
||
|
|
String(50), nullable=False, default="other"
|
||
|
|
) # design | implementation | test | docs | dependencies | performance | security | other
|
||
|
|
severity: Mapped[str] = mapped_column(String(20), nullable=False, default="medium")
|
||
|
|
status: Mapped[TDStatus] = mapped_column(
|
||
|
|
Enum(TDStatus, name="tdstatus"), nullable=False, default=TDStatus.open
|
||
|
|
)
|
||
|
|
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
|
||
|
|
)
|
||
|
|
|
||
|
|
topic: Mapped["Topic"] = relationship("Topic", lazy="selectin") # noqa: F821
|
||
|
|
workstream: Mapped["Workstream"] = relationship("Workstream", lazy="selectin") # noqa: F821
|