state-hub/api/schemas/extension_point.py

import uuid
from datetime import datetime

from pydantic import AliasChoices, BaseModel, ConfigDict, Field

from api.models.extension_point import EPStatus

VALID_PRIORITIES = {"low", "medium", "high", "critical"}


class EPCreate(BaseModel):
    ep_id: str | None = None
    domain: str  # slug; router resolves to domain_id FK
    title: str
    description: str | None = None
    location: str | None = None
    ep_type: str = "other"
    status: EPStatus = EPStatus.open
    priority: str = "medium"
    topic_id: uuid.UUID | None = None
    workplan_id: uuid.UUID | None = Field(
        default=None,
        validation_alias=AliasChoices("workplan_id", "workstream_id"),
    )


class EPUpdate(BaseModel):
    ep_id: str | None = None
    title: str | None = None
    description: str | None = None
    location: str | None = None
    ep_type: str | None = None
    status: EPStatus | None = None
    priority: str | None = None
    workplan_id: uuid.UUID | None = Field(
        default=None,
        validation_alias=AliasChoices("workplan_id", "workstream_id"),
    )


class EPRead(BaseModel):
    model_config = ConfigDict(from_attributes=True)

    id: uuid.UUID
    ep_id: str | None = None
    domain_slug: str  # derived from domain relationship
    title: str
    description: str | None = None
    location: str | None = None
    ep_type: str
    status: EPStatus
    priority: str
    topic_id: uuid.UUID | None = None
    workplan_id: uuid.UUID | None = None
    created_at: datetime

    @property
    def workstream_id(self) -> uuid.UUID | None:
        return self.workplan_id
    updated_at: datetime