66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from app.models.task import TaskPriority, TaskStatus
|
|
|
|
|
|
class TaskBase(BaseModel):
|
|
title: str = Field(..., min_length=1, max_length=100)
|
|
description: Optional[str] = None
|
|
priority: TaskPriority = TaskPriority.MEDIUM
|
|
status: TaskStatus = TaskStatus.TODO
|
|
due_date: Optional[datetime] = None
|
|
completed: bool = False
|
|
|
|
model_config = {
|
|
"json_encoders": {
|
|
datetime: lambda dt: dt.isoformat(),
|
|
}
|
|
}
|
|
|
|
|
|
class TaskCreate(TaskBase):
|
|
pass
|
|
|
|
|
|
class TaskUpdate(BaseModel):
|
|
title: Optional[str] = Field(None, min_length=1, max_length=100)
|
|
description: Optional[str] = None
|
|
priority: Optional[TaskPriority] = None
|
|
status: Optional[TaskStatus] = None
|
|
due_date: Optional[datetime] = None
|
|
completed: Optional[bool] = None
|
|
|
|
model_config = {
|
|
"json_encoders": {
|
|
datetime: lambda dt: dt.isoformat(),
|
|
},
|
|
"populate_by_name": True,
|
|
"json_schema_extra": {
|
|
"examples": [
|
|
{
|
|
"title": "Updated Task Title",
|
|
"description": "Updated task description",
|
|
"priority": "high",
|
|
"status": "in_progress",
|
|
"completed": False
|
|
}
|
|
]
|
|
}
|
|
}
|
|
|
|
|
|
class TaskInDBBase(TaskBase):
|
|
id: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {
|
|
"from_attributes": True
|
|
}
|
|
|
|
|
|
class Task(TaskInDBBase):
|
|
pass |