52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
|
import uuid
|
||
|
|
from datetime import date, datetime
|
||
|
|
|
||
|
|
from pydantic import BaseModel, ConfigDict, model_validator
|
||
|
|
|
||
|
|
from api.models.task import TaskPriority, TaskStatus
|
||
|
|
|
||
|
|
|
||
|
|
class TaskCreate(BaseModel):
|
||
|
|
workstream_id: uuid.UUID
|
||
|
|
title: str
|
||
|
|
description: str | None = None
|
||
|
|
status: TaskStatus = TaskStatus.todo
|
||
|
|
priority: TaskPriority = TaskPriority.medium
|
||
|
|
assignee: str | None = None
|
||
|
|
due_date: date | None = None
|
||
|
|
blocking_reason: str | None = None
|
||
|
|
parent_task_id: uuid.UUID | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class TaskUpdate(BaseModel):
|
||
|
|
title: str | None = None
|
||
|
|
description: str | None = None
|
||
|
|
status: TaskStatus | None = None
|
||
|
|
priority: TaskPriority | None = None
|
||
|
|
assignee: str | None = None
|
||
|
|
due_date: date | None = None
|
||
|
|
blocking_reason: str | None = None
|
||
|
|
parent_task_id: uuid.UUID | None = None
|
||
|
|
|
||
|
|
@model_validator(mode="after")
|
||
|
|
def blocking_reason_required_when_blocked(self) -> "TaskUpdate":
|
||
|
|
if self.status == TaskStatus.blocked and not self.blocking_reason:
|
||
|
|
raise ValueError("blocking_reason is required when status is blocked")
|
||
|
|
return self
|
||
|
|
|
||
|
|
|
||
|
|
class TaskRead(BaseModel):
|
||
|
|
model_config = ConfigDict(from_attributes=True)
|
||
|
|
id: uuid.UUID
|
||
|
|
workstream_id: uuid.UUID
|
||
|
|
title: str
|
||
|
|
description: str | None = None
|
||
|
|
status: TaskStatus
|
||
|
|
priority: TaskPriority
|
||
|
|
assignee: str | None = None
|
||
|
|
due_date: date | None = None
|
||
|
|
blocking_reason: str | None = None
|
||
|
|
parent_task_id: uuid.UUID | None = None
|
||
|
|
created_at: datetime
|
||
|
|
updated_at: datetime
|