"""Input schemas for price calculation."""

from enum import StrEnum

from pydantic import BaseModel, Field, StrictInt


class DeliveryEntity(StrEnum):
    INDIVIDUAL = "individual"
    LEGAL = "legal"


class ParcelType(StrEnum):
    DOC = "doc"
    PARCEL = "parcel"


class DeliveryCalculationRequest(BaseModel):
    entity: DeliveryEntity
    from_city: StrictInt
    to_city: StrictInt
    weight_kg: float = Field(gt=0)
    length_cm: float = Field(gt=0)
    width_cm: float = Field(gt=0)
    height_cm: float = Field(gt=0)
    parcel_type: ParcelType | None = None