Раис Юсупалиев
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
"""CDEK order registration payload mappers."""
|
|
|
|
from typing import Any
|
|
|
|
from app.schemas.payment import (
|
|
InitPaymentRequest,
|
|
PaymentPackage,
|
|
PaymentParty,
|
|
)
|
|
|
|
|
|
class CDEKOrderMappingError(ValueError):
|
|
"""Raised when CDEK order payload cannot be mapped."""
|
|
|
|
|
|
def map_cdek_order_request(request: InitPaymentRequest) -> dict[str, Any]:
|
|
payload: dict[str, Any] = {
|
|
"type": request.type,
|
|
"tariff_code": request.tariff_code,
|
|
"sender": _map_order_party(request.sender),
|
|
"recipient": _map_order_party(request.recipient),
|
|
"from_location": request.from_location.model_dump(mode="python"),
|
|
"to_location": request.to_location.model_dump(mode="python"),
|
|
"packages": [_map_order_package(package) for package in request.packages],
|
|
}
|
|
if request.comment is not None:
|
|
payload["comment"] = request.comment
|
|
if request.services is not None:
|
|
payload["services"] = [
|
|
service.model_dump(mode="python") for service in request.services
|
|
]
|
|
return payload
|
|
|
|
|
|
def map_cdek_order_response(payload: dict[str, Any]) -> str:
|
|
entity = payload.get("entity")
|
|
if not isinstance(entity, dict):
|
|
raise CDEKOrderMappingError("CDEK order response must include entity object.")
|
|
|
|
order_uuid = entity.get("uuid")
|
|
if not isinstance(order_uuid, str) or not order_uuid:
|
|
raise CDEKOrderMappingError("CDEK order response must include entity.uuid.")
|
|
|
|
return order_uuid
|
|
|
|
|
|
def _map_order_party(party: PaymentParty) -> dict[str, Any]:
|
|
return {
|
|
"name": party.name,
|
|
"email": party.email,
|
|
"phones": [party.phone.model_dump(mode="python")],
|
|
}
|
|
|
|
|
|
def _map_order_package(package: PaymentPackage) -> dict[str, Any]:
|
|
payload = package.model_dump(mode="python", exclude_none=True)
|
|
payload["weight"] = package.weight * 1000
|
|
return payload
|