o h*@sDUdZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZddl mZddlmZmZmZd d lmZd d lmZmZd d lmZejd?iejd diGdddZejd?iejd diGdddZere e eeefe!eefeefZ"de#d<e ej$e"fZ%e ej&e"fZ'e de%dZ(e de'dZ)e ddddd@d%d&Z*e dddddAd)d&Z*e dddddBd,d&Z*d-ed.dd/dCd2d&Z*e d3e defdZ+e dDd5d6Z,e dd.dd7dEd9d6Z, dFd-d.ed7dGd;d6Z,e d<Z-eree-dfZ.dSejd?iejGd=d>d>Z.dS)HzEThis module contains related classes and functions for serialization.) annotationsN) partialmethod) TYPE_CHECKINGAnyCallableTypeVarUnionoverload)PydanticUndefined core_schema)r ) AnnotatedLiteral TypeAlias)PydanticUndefinedAnnotation) _decorators_internal_dataclass)GetCoreSchemaHandlerfrozenTc@<eZdZUdZded<eZded<dZded<dddZdS)PlainSerializeraPlain serializers use a function to modify the output of serialization. Attributes: func: The serializer function. return_type: The return type for the function. If omitted it will be inferred from the type annotation. when_used: Determines when this serializer should be used. Accepts a string with values `'always'`, `'unless-none'`, `'json'`, and `'json-unless-none'`. Defaults to 'always'. zcore_schema.SerializerFunctionfuncr return_typealwaysCallable[[_WrapSerializeMethodType], _WrapSerializeMethodType]cGrJrKr2rOr2r2r3rLrPr!r)rMrrrDLiteral['plain', 'wrap']Callable[[Any], Any]csdfdd }|S) a{Decorator that enables custom field serialization. See [Custom serializers](../concepts/serialization.md#custom-serializers) for more information. Four signatures are supported: - `(self, value: Any, info: FieldSerializationInfo)` - `(self, value: Any, nxt: SerializerFunctionWrapHandler, info: FieldSerializationInfo)` - `(value: Any, info: SerializationInfo)` - `(value: Any, nxt: SerializerFunctionWrapHandler, info: SerializationInfo)` Args: fields: Which field(s) the method should be called on. mode: The serialization mode. - `plain` means the function will be called instead of the default serialization logic, - `wrap` means the function will be called with an argument to optionally call the default serialization logic. return_type: Optional return type for the function, if omitted it will be inferred from the type annotation. when_used: Determines the serializer will be used for serialization. check_fields: Whether to check that the fields actually exist on the model. Returns: The decorator function. fHCallable[..., Any] | staticmethod[Any, Any] | classmethod[Any, Any, Any]r(_decorators.PydanticDescriptorProxy[Any]cs tjd}t||S)N)rGrMrrrD)rFieldSerializerDecoratorInfoPydanticDescriptorProxyrUdec_inforDrGrMrrr2r3decs zfield_serializer..decN)rUrVrrWr2)rMrrrDrGr]r2r\r3rLs! FuncType__fcCrJrKr2)r_r2r2r3model_serializersr`rMrrCallable[[FuncType], FuncType]cCrJrKr2rar2r2r3r`sCallable[..., Any] | Nonecs&dfdd }|dur|S||S) aDecorator that enables custom model serialization. See [Custom serializers](../concepts/serialization.md#custom-serializers) for more information. Args: __f: The function to be decorated. mode: The serialization mode. - `'plain'` means the function will be called instead of the default serialization logic - `'wrap'` means the function will be called with an argument to optionally call the default serialization logic. when_used: Determines when this serializer should be used. return_type: The return type for the function. If omitted it will be inferred from the type annotation. Returns: The decorator function. rUCallable[..., Any]rrWcstjd}t||S)NrMrr)rModelSerializerDecoratorInforYrZrer2r3r]s zmodel_serializer..decN)rUrdrrWr2)r_rMrrr]r2rer3r`sAnyTypec@s&eZdZd ddZdd d ZejZd S)SerializeAsAnyitemrrcCst|tfSrK)r rh)clsrir2r2r3__class_getitem__sz SerializeAsAny.__class_getitem__rrrrcCsR||}|}|ddkr|}|d}|ddks tjddtd|d<|S)Ntype definitionsr0cSs||SrKr2)xhr2r2r3sz=SerializeAsAny.__get_pydantic_core_schema__..)r0r&)copyr r? any_schema)r/rrr0schema_to_updater2r2r3r4s    z+SerializeAsAny.__get_pydantic_core_schema__N)rirrrr6)r8r9r:rkr4object__hash__r2r2r2r3rhs   rhr2) rErFrGrFrrrrrDrHrrI)rErFrGrFrMrNrrrrrDrHrrI)rErFrGrFrMrQrrrrrDrHrrR) rGrFrMrSrrrrrDrHrrT)r_r^rr^)rMrSrrrrrrbrK) r_rcrMrSrrrrrrT)/r; __future__r dataclasses functoolsrtypingrrrrrr pydantic_corer r _core_schematyping_extensionsr r rr _internalrrannotated_handlersr dataclass slots_truerr= classmethod staticmethodr@r<SerializerFunction_PlainSerializationFunctionWrapSerializerFunction_WrapSerializationFunctionrArCrLr^r`rgrhr2r2r2r3sz      )*,     0  #