Исходный код cdek.apps.models.entity_response

from __future__ import annotations

from uuid import UUID

from pydantic import BaseModel, Field

from .related_entity import RelatedEntity
from .request import Request




class RootEntity(BaseModel):
    uuid: UUID = Field(..., description="Идентификатор сущности в ИС СДЭК")





class EntityResponse(BaseModel):
    """Модель ответа о сущности."""

    entity: RootEntity | None = Field(
        default=None, description="Идентификатор сущности в ИС СДЭК"
    )
    requests: list[Request] | None = None
    related_entities: list[RelatedEntity] | None = None



    def get_entry_uuid(self) -> str | None:
        """Получить UUID сущности."""
        if self.entity is None:
            return None
        if hasattr(self.entity, "uuid"):
            return str(self.entity.uuid)
        return None




    def get_entity(self) -> RootEntity | None:
        """Получить данные сущности."""
        if self.entity is None:
            return None
        return self.entity




    def get_requests(self) -> list[Request]:
        """Получить список запросов."""
        if self.requests is None:
            return []
        return self.requests