Раис Юсупалиев
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
"""Schemas for CDEK order registration."""
|
|
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field, model_validator
|
|
|
|
|
|
class OrderPhone(BaseModel):
|
|
number: str = Field(min_length=1)
|
|
|
|
|
|
class OrderParty(BaseModel):
|
|
name: str = Field(min_length=1)
|
|
email: str = Field(min_length=1)
|
|
phones: list[OrderPhone] = Field(min_length=1)
|
|
|
|
@model_validator(mode="before")
|
|
@classmethod
|
|
def reject_company_field(cls, value: object) -> object:
|
|
if isinstance(value, dict) and "company" in value:
|
|
raise ValueError("company is not allowed")
|
|
return value
|
|
|
|
|
|
class OrderLocation(BaseModel):
|
|
address: str = Field(min_length=1)
|
|
city: str = Field(min_length=1)
|
|
country_code: str = Field(min_length=2, max_length=2)
|
|
|
|
|
|
class OrderService(BaseModel):
|
|
code: str = Field(min_length=1)
|
|
parameter: str = Field(min_length=1)
|
|
|
|
|
|
class OrderPackage(BaseModel):
|
|
number: str = Field(min_length=1)
|
|
weight: int = Field(gt=0)
|
|
length: int = Field(gt=0)
|
|
width: int = Field(gt=0)
|
|
height: int = Field(gt=0)
|
|
comment: str | None = None
|
|
|
|
|
|
class OrderCreateRequest(BaseModel):
|
|
type: Literal[2]
|
|
tariff_code: Literal[535]
|
|
comment: str | None = None
|
|
sender: OrderParty
|
|
recipient: OrderParty
|
|
from_location: OrderLocation
|
|
to_location: OrderLocation
|
|
services: list[OrderService] = Field(min_length=1)
|
|
packages: list[OrderPackage] = Field(min_length=1)
|
|
|
|
|
|
class OrderCreateResponse(BaseModel):
|
|
provider: str = Field(min_length=1)
|
|
order_uuid: str = Field(min_length=1)
|