Module aws_lambda_powertools.utilities.idempotency.serialization.pydantic

Classes

class PydanticSerializer (model: Type[pydantic.main.BaseModel])

Pydantic serializer for idempotency models

Parameters

model : Model

Pydantic model to be used for serialization

Expand source code

class PydanticSerializer(BaseIdempotencyModelSerializer):
    """Pydantic serializer for idempotency models"""

    def __init__(self, model: Type[BaseModel]):
        """
        Parameters
        ----------
        model: Model
            Pydantic model to be used for serialization
        """
        self.__model: Type[BaseModel] = model

    def to_dict(self, data: BaseModel) -> Dict:
        if callable(getattr(data, "model_dump", None)):
            # Support for pydantic V2
            return data.model_dump()  # type: ignore[unused-ignore,attr-defined]
        return data.dict()

    def from_dict(self, data: Dict) -> BaseModel:
        if callable(getattr(self.__model, "model_validate", None)):
            # Support for pydantic V2
            return self.__model.model_validate(data)  # type: ignore[unused-ignore,attr-defined]
        return self.__model.parse_obj(data)

    @classmethod
    def instantiate(cls, model_type: Any) -> BaseIdempotencySerializer:
        if model_type is None:
            raise IdempotencyNoSerializationModelError("No serialization model was supplied")

        if not issubclass(model_type, BaseModel):
            raise IdempotencyModelTypeError("Model type is not inherited from pydantic BaseModel")

        return cls(model=model_type)

Ancestors

• BaseIdempotencyModelSerializer
• BaseIdempotencySerializer
• abc.ABC

Methods

def from_dict(self, data: Dict) ‑> pydantic.main.BaseModel

def to_dict(self, data: pydantic.main.BaseModel) ‑> Dict

Inherited members

• BaseIdempotencyModelSerializer:
  • instantiate