55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
from datetime import datetime
|
|
from enum import Enum
|
|
from typing import Optional
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class TaskStatus(str, Enum):
|
|
"""Valid status values for tasks."""
|
|
PENDING = "pending"
|
|
IN_PROGRESS = "in_progress"
|
|
COMPLETED = "completed"
|
|
CANCELLED = "cancelled"
|
|
|
|
|
|
class TaskPriority(str, Enum):
|
|
"""Valid priority values for tasks."""
|
|
LOW = "low"
|
|
MEDIUM = "medium"
|
|
HIGH = "high"
|
|
URGENT = "urgent"
|
|
|
|
|
|
class TaskBase(BaseModel):
|
|
"""Base Task schema with shared attributes."""
|
|
title: str
|
|
description: Optional[str] = None
|
|
status: TaskStatus = TaskStatus.PENDING
|
|
priority: TaskPriority = TaskPriority.MEDIUM
|
|
due_date: Optional[datetime] = None
|
|
completed: Optional[bool] = False
|
|
|
|
|
|
class TaskCreate(TaskBase):
|
|
"""Schema for creating a new task."""
|
|
pass
|
|
|
|
|
|
class TaskUpdate(BaseModel):
|
|
"""Schema for updating an existing task."""
|
|
title: Optional[str] = None
|
|
description: Optional[str] = None
|
|
status: Optional[TaskStatus] = None
|
|
priority: Optional[TaskPriority] = None
|
|
due_date: Optional[datetime] = None
|
|
completed: Optional[bool] = None
|
|
|
|
|
|
class TaskResponse(TaskBase):
|
|
"""Schema for task responses with database fields."""
|
|
id: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True |