src/tasks/schema.py - main - evlx/task_tracker

src/tasks/schema.py
72 строки2 KB
evlx
удалены лишние комментарии
29 май 2024, 13:24
29 май 2024, 13:24712f5eb
from datetime import datetime
from typing import List
from uuid import UUID

from pydantic import BaseModel


class BaseTaskSchema(BaseModel):
    """
    Базовая схема Задание.
    Без UUID

    Attributes:
    -----------
        name : str  Название задания.
        content : str   Описание задания.
        period_of_execution : Optional[datetime]    Период выполнения задания.
        parent_id : Optional[UUID]  ID родительского задания.
        status : int    Статус задания.
        employee_id : Optional[UUID]    ID сотрудника.
    """
    name: str
    content: str
    period_of_execution: datetime | None = None
    parent_id: UUID | None = None
    status: int = 0
    employee_id: UUID | None = None

    class Config:
        """
        Настройки схемы
        """
        from_attributes = True
        population_by_name = True
        arbitrary_types_allowed = True
        json_schema_extra = {
            "example": {
                "name": "Задание1",
                "content": "Описание задания",
                "period_of_execution": "2024-04-28 19:35:00+03:00",
                "status": "0"
            }
        }


class TaskSchema(BaseTaskSchema):
    """
    Схема Задание.
    Добавлен UUID

    Attributes:
    -----------
        id : UUID   ID задания.
    """
    id: UUID  # ID задания


class TaskCreateUpdateSchema(BaseTaskSchema):
    """
    Схема Задание для создания/обновления.
    """


class TasksList(BaseModel):
    """
    Список заданий

    Attributes:
    -----------
        tasks : List[TaskSchema]
    """
    tasks: List[TaskSchema]