a y{Jh* @sUdZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZddl mZddlmZmZmZd d lmZd d lmZmZd d lmZejfiejd diGdddZejfiejd diGdddZerTe e eeefe!eefeefZ"de#d<e ej$e"fZ%e ej&e"fZ'e de%dZ(e de'dZ)e dddddddddddd d!Z*e ddddddd"ddddd#d$d!Z*e ddddddd%dddd&d#d'd!Z*d(ed)dd*dd+dddd,d-d.d!Z*e d/e defdZ+e d/d/d0d1d2Z,e dd)dd3d+ddd4d5d6d2Z,d=d(d)ed3d7d+ddd,d8d9d2Z,e d:Z-ervee-dfZ.n ejfiejGd;d<d<Z.dS)>zEThis module contains related classes and functions for serialization.) annotationsN) partialmethod) TYPE_CHECKINGAnyCallableTypeVarUnionoverload)PydanticUndefined core_schema)r ) AnnotatedLiteral TypeAlias)PydanticUndefinedAnnotation) _decorators_internal_dataclass)GetCoreSchemaHandlerfrozenTc@sDeZdZUdZded<eZded<dZded<dd d d d d ZdS)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]cGsdSrCr/rFr/r/r0rDs r r)rErrr?zLiteral['plain', 'wrap']zCallable[[Any], Any])rBrErrr?rcs dddfdd }|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. zHCallable[..., Any] | staticmethod[Any, Any] | classmethod[Any, Any, Any](_decorators.PydanticDescriptorProxy[Any]frcs tjd}t||S)N)rBrErrr?)rZFieldSerializerDecoratorInfoPydanticDescriptorProxyrIZdec_infor?rBrErrr/r0decszfield_serializer..decr/)rErrr?rBrMr/rLr0rDs! FuncType)__frcCsdSrCr/)rOr/r/r0model_serializersrPrErrzCallable[[FuncType], FuncType])rErrrcCsdSrCr/rQr/r/r0rPszCallable[..., Any] | None)rOrErrrcs0dddfdd }|dur$|S||SdS)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. zCallable[..., Any]rGrHcstjd}t||S)NrErr)rZModelSerializerDecoratorInforJrKrRr/r0rMszmodel_serializer..decNr/)rOrErrrMr/rRr0rPsAnyTypec@s4eZdZdddddZdddddd ZejZd S) SerializeAsAnyr)itemrcCst|tfSrC)r rT)clsrUr/r/r0__class_getitem__sz SerializeAsAny.__class_getitem__rrrcCsH||}|}|ddkr*|}|d}q tjddtd|d<|S)NtypeZ definitionsr-cSs||SrCr/)xhr/r/r0z=SerializeAsAny.__get_pydantic_core_schema__..)r-r$)copyr r:Z any_schema)r,rrr-Zschema_to_updater/r/r0r1s   z+SerializeAsAny.__get_pydantic_core_schema__N)r3r4r5rWr1object__hash__r/r/r/r0rTs rT)N)/r6 __future__r dataclasses functoolsrtypingrrrrrr Z pydantic_corer r Z _core_schematyping_extensionsr r rr _internalrrZannotated_handlersr dataclassZ slots_truerr8 classmethod staticmethodr;r7ZSerializerFunctionZ_PlainSerializationFunctionZWrapSerializerFunctionZ_WrapSerializationFunctionr<r>rDrNrPrSrTr/r/r/r0sx      )*,     0 #