schemas.py

from decimal import Decimal
from datetime import datetime
from typing import List, Optional

from pydantic import BaseModel, Field, conint


class AddOrderItemRequest(BaseModel):
    """Тело запроса: добавление товара в заказ."""

    nomenclature_id: int = Field(..., description="ID номенклатуры")
    quantity: conint(gt=0) = Field(..., description="Количество")

    model_config = {"json_schema_extra": {"example": {"nomenclature_id": 1, "quantity": 2}}}


class RemoveOrderItemRequest(BaseModel):
    """Тело запроса: удаление товара из заказа.

    Можно удалить позицию полностью или уменьшить количество.
    При удалении товар возвращается в остатки по номенклатуре.
    """

    nomenclature_id: int = Field(..., description="ID номенклатуры в заказе")
    quantity: Optional[conint(gt=0)] = Field(
        None,
        description=(
            "Количество для удаления. "
            "Если не указано или больше/равно текущему количеству в заказе, "
            "позиция будет удалена полностью."
        ),
    )

    model_config = {
        "json_schema_extra": {
            "example": {
                "nomenclature_id": 1,
                "quantity": 1,
            }
        }
    }


class OrderItemResponse(BaseModel):
    """Позиция заказа в ответе API."""

    id: int
    order_id: int
    nomenclature_id: int
    quantity: int
    price: Decimal

    model_config = {"from_attributes": True}


class OrderCreate(BaseModel):
    """Тело запроса: создание заказа."""

    client_id: int = Field(..., description="ID клиента")

    model_config = {
        "json_schema_extra": {
            "example": {
                "client_id": 1,
            }
        }
    }


class OrderResponse(BaseModel):
    """Информация о заказе."""

    id: int
    client_id: int
    created_at: datetime
    items: List[OrderItemResponse] = []

    model_config = {"from_attributes": True}


class NomenclatureResponse(BaseModel):
    """Номенклатура в ответе API."""

    id: int
    name: str
    quantity: int
    price: Decimal
    category_id: Optional[int] = None

    model_config = {"from_attributes": True}


class ErrorResponse(BaseModel):
    """Единообразный формат ошибки."""

    detail: str = Field(..., description="Описание ошибки")
    code: Optional[str] = Field(None, description="Код ошибки")