100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
"""Schemas for task CRUD and task comment API payloads."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Literal, Self
|
|
from uuid import UUID
|
|
|
|
from pydantic import field_validator, model_validator
|
|
from sqlmodel import Field, SQLModel
|
|
|
|
from app.schemas.common import NonEmptyStr
|
|
from app.schemas.task_tags import TaskTagRef
|
|
|
|
TaskStatus = Literal["inbox", "in_progress", "review", "done"]
|
|
STATUS_REQUIRED_ERROR = "status is required"
|
|
# Keep these symbols as runtime globals so Pydantic can resolve
|
|
# deferred annotations reliably.
|
|
RUNTIME_ANNOTATION_TYPES = (datetime, UUID, NonEmptyStr, TaskTagRef)
|
|
|
|
|
|
class TaskBase(SQLModel):
|
|
"""Shared task fields used by task create/read payloads."""
|
|
|
|
title: str
|
|
description: str | None = None
|
|
status: TaskStatus = "inbox"
|
|
priority: str = "medium"
|
|
due_at: datetime | None = None
|
|
assigned_agent_id: UUID | None = None
|
|
depends_on_task_ids: list[UUID] = Field(default_factory=list)
|
|
tag_ids: list[UUID] = Field(default_factory=list)
|
|
|
|
|
|
class TaskCreate(TaskBase):
|
|
"""Payload for creating a task."""
|
|
|
|
created_by_user_id: UUID | None = None
|
|
|
|
|
|
class TaskUpdate(SQLModel):
|
|
"""Payload for partial task updates."""
|
|
|
|
title: str | None = None
|
|
description: str | None = None
|
|
status: TaskStatus | None = None
|
|
priority: str | None = None
|
|
due_at: datetime | None = None
|
|
assigned_agent_id: UUID | None = None
|
|
depends_on_task_ids: list[UUID] | None = None
|
|
tag_ids: list[UUID] | None = None
|
|
comment: NonEmptyStr | None = None
|
|
|
|
@field_validator("comment", mode="before")
|
|
@classmethod
|
|
def normalize_comment(cls, value: object) -> object | None:
|
|
"""Normalize blank comment strings to `None`."""
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, str) and not value.strip():
|
|
return None
|
|
return value
|
|
|
|
@model_validator(mode="after")
|
|
def validate_status(self) -> Self:
|
|
"""Ensure explicitly supplied status is not null."""
|
|
if "status" in self.model_fields_set and self.status is None:
|
|
raise ValueError(STATUS_REQUIRED_ERROR)
|
|
return self
|
|
|
|
|
|
class TaskRead(TaskBase):
|
|
"""Task payload returned from read endpoints."""
|
|
|
|
id: UUID
|
|
board_id: UUID | None
|
|
created_by_user_id: UUID | None
|
|
in_progress_at: datetime | None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
blocked_by_task_ids: list[UUID] = Field(default_factory=list)
|
|
is_blocked: bool = False
|
|
tags: list[TaskTagRef] = Field(default_factory=list)
|
|
|
|
|
|
class TaskCommentCreate(SQLModel):
|
|
"""Payload for creating a task comment."""
|
|
|
|
message: NonEmptyStr
|
|
|
|
|
|
class TaskCommentRead(SQLModel):
|
|
"""Task comment payload returned from read endpoints."""
|
|
|
|
id: UUID
|
|
message: str | None
|
|
agent_id: UUID | None
|
|
task_id: UUID | None
|
|
created_at: datetime
|