108 lines
2.3 KiB
Python
108 lines
2.3 KiB
Python
from pydantic import BaseModel, ConfigDict
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
from .models import DEFAULT_STATUSES
|
|
|
|
|
|
# Task Schemas
|
|
class TaskBase(BaseModel):
|
|
title: str
|
|
description: Optional[str] = None
|
|
status: str = "backlog"
|
|
parent_task_id: Optional[int] = None
|
|
sort_order: int = 0
|
|
estimated_minutes: Optional[int] = None
|
|
tags: Optional[List[str]] = None
|
|
flag_color: Optional[str] = None
|
|
|
|
|
|
class TaskCreate(TaskBase):
|
|
project_id: int
|
|
|
|
|
|
class TaskUpdate(BaseModel):
|
|
title: Optional[str] = None
|
|
description: Optional[str] = None
|
|
status: Optional[str] = None
|
|
parent_task_id: Optional[int] = None
|
|
sort_order: Optional[int] = None
|
|
estimated_minutes: Optional[int] = None
|
|
tags: Optional[List[str]] = None
|
|
flag_color: Optional[str] = None
|
|
|
|
|
|
class Task(TaskBase):
|
|
id: int
|
|
project_id: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
class TaskWithSubtasks(Task):
|
|
subtasks: List['TaskWithSubtasks'] = []
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
# Project Schemas
|
|
class ProjectBase(BaseModel):
|
|
name: str
|
|
description: Optional[str] = None
|
|
|
|
|
|
class ProjectCreate(ProjectBase):
|
|
statuses: Optional[List[str]] = None
|
|
|
|
|
|
class ProjectUpdate(BaseModel):
|
|
name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
statuses: Optional[List[str]] = None
|
|
is_archived: Optional[bool] = None
|
|
|
|
|
|
class Project(ProjectBase):
|
|
id: int
|
|
statuses: List[str]
|
|
is_archived: bool
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
class ProjectWithTasks(Project):
|
|
tasks: List[Task] = []
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
# JSON Import Schemas
|
|
class ImportSubtask(BaseModel):
|
|
title: str
|
|
description: Optional[str] = None
|
|
status: str = "backlog"
|
|
estimated_minutes: Optional[int] = None
|
|
tags: Optional[List[str]] = None
|
|
flag_color: Optional[str] = None
|
|
subtasks: List['ImportSubtask'] = []
|
|
|
|
|
|
class ImportProject(BaseModel):
|
|
name: str
|
|
description: Optional[str] = None
|
|
statuses: Optional[List[str]] = None
|
|
|
|
|
|
class ImportData(BaseModel):
|
|
project: ImportProject
|
|
tasks: List[ImportSubtask] = []
|
|
|
|
|
|
class ImportResult(BaseModel):
|
|
project_id: int
|
|
project_name: str
|
|
tasks_created: int
|