a hB @sUdZddlmZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZmZmZddlmZd d lmZd d lmZmZd d lmZejfiej d diGdddZ!ejfiej d diGdddZ"erDdZ#de$d<dZ%de$d<dZ&de$d<dZ'de$d<e de%dZ(e de&dZ)edddd d!d!d"d#d$d%d&d'd(d)Z*eddddd*d!d!d+d#d$d%d,d'd-d)Z*d.ed/dd*d!d0d#d$d%d1d2d3d)Z*er\e e ege fZ+de$d4<e e ge fZ,de$d5<d6Z-de$d7<e e eege fZ.de$d8<e e ege fZ/de$d9<d:Z0de$d;<de-dZ2e d?e0dZ3ed>d>d@dAdBZ4ed/ddCd"d$d#dDdEdFdBZ4edd/ddGd+d$d#dHdEdIdBZ4dQd.d/edGdJd0d$d#dKdLdMdBZ4e dNZ5ere e5dfZ6n ejfiej GdOdPdPZ6dS)RzEThis module contains related classes and functions for serialization.) annotationsN)partial partialmethod) TYPE_CHECKING AnnotatedAnyCallableLiteralTypeVaroverload)PydanticUndefined core_schema)SerializationInfoSerializerFunctionWrapHandlerWhenUsed) 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. This is particularly helpful when you want to customize the serialization for annotated types. Consider an input of `list`, which will be serialized into a space-delimited string. ```python from typing import Annotated from pydantic import BaseModel, PlainSerializer CustomStr = Annotated[ list, PlainSerializer(lambda x: ' '.join(x), return_type=str) ] class StudentModel(BaseModel): courses: CustomStr student = StudentModel(courses=['Math', 'Chemistry', 'English']) print(student.model_dump()) #> {'courses': 'Math Chemistry English'} ``` 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_typealwaysr when_usedrcore_schema.CoreSchema source_typehandlerreturnc Cs||}|jtur|j}nLztj|j|jd}Wn0tyd}zt ||WYd}~n d}~00|turrdn| |}t j |jt |jd||jd|d<|S)zGets the Pydantic core schema. Args: source_type: The source type. handler: The `GetCoreSchemaHandler` instance. Returns: The Pydantic core schema. localnsNplainfunctionZinfo_arg return_schemar serialization)rr rget_callable_return_typer_get_types_namespacelocals NameErrorrfrom_name_errorgenerate_schemar Z$plain_serializer_function_ser_schemainspect_annotated_serializerrselfrr schemarer'r4[/var/www/html/assistant/venv/lib/python3.9/site-packages/pydantic/functional_serializers.py__get_pydantic_core_schema__5s$   "  z,PlainSerializer.__get_pydantic_core_schema__N __name__ __module__ __qualname____doc____annotations__r rrr6r4r4r4r5rs   rc@sDeZdZUdZded<eZded<dZded<dd d d d d ZdS)WrapSerializeraWrap serializers receive the raw inputs along with a handler function that applies the standard serialization logic, and can modify the resulting value before returning it as the final output of serialization. For example, here's a scenario in which a wrap serializer transforms timezones to UTC **and** utilizes the existing `datetime` serialization logic. ```python from datetime import datetime, timezone from typing import Annotated, Any from pydantic import BaseModel, WrapSerializer class EventDatetime(BaseModel): start: datetime end: datetime def convert_to_utc(value: Any, handler, info) -> dict[str, datetime]: # Note that `handler` can actually help serialize the `value` for # further custom serialization in case it's a subclass. partial_result = handler(value, info) if info.mode == 'json': return { k: datetime.fromisoformat(v).astimezone(timezone.utc) for k, v in partial_result.items() } return {k: v.astimezone(timezone.utc) for k, v in partial_result.items()} UTCEventDatetime = Annotated[EventDatetime, WrapSerializer(convert_to_utc)] class EventModel(BaseModel): event_datetime: UTCEventDatetime dt = EventDatetime( start='2024-01-01T07:00:00-08:00', end='2024-01-03T20:00:00+06:00' ) event = EventModel(event_datetime=dt) print(event.model_dump()) ''' { 'event_datetime': { 'start': datetime.datetime( 2024, 1, 1, 15, 0, tzinfo=datetime.timezone.utc ), 'end': datetime.datetime( 2024, 1, 3, 14, 0, tzinfo=datetime.timezone.utc ), } } ''' print(event.model_dump_json()) ''' {"event_datetime":{"start":"2024-01-01T15:00:00Z","end":"2024-01-03T14:00:00Z"}} ''' ``` Attributes: func: The serializer function to be wrapped. 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'. z"core_schema.WrapSerializerFunctionrrrrrrrrrc Cs||}|jtur|j}nLztj|j|jd}Wn0tyd}zt ||WYd}~n d}~00|turrdn| |}t j |jt |jd||jd|d<|S)zThis method is used to get the Pydantic core schema of the class. Args: source_type: Source type. handler: Core schema handler. Returns: The generated core schema of the class. r"Nwrapr%r()rr rr)rr*r+r,rr-r.r #wrap_serializer_function_ser_schemar/rr0r4r4r5r6s$   "  z+WrapSerializer.__get_pydantic_core_schema__Nr7r4r4r4r5r=Xs >  r=z!partial[Any] | partialmethod[Any]r_Partialz)core_schema.SerializerFunction | _PartialFieldPlainSerializerz-core_schema.WrapSerializerFunction | _PartialFieldWrapSerializerz*FieldPlainSerializer | FieldWrapSerializerFieldSerializer_FieldPlainSerializerT)bound_FieldWrapSerializerT.)rr check_fieldsstrzLiteral['wrap']rrz bool | Nonez8Callable[[_FieldWrapSerializerT], _FieldWrapSerializerT])fieldfieldsmoderrrGr!cGsdSNr4rIrKrrrGrJr4r4r5field_serializers rN)rKrrrGzLiteral['plain']z:Callable[[_FieldPlainSerializerT], _FieldPlainSerializerT]cGsdSrLr4rMr4r4r5rNs r$rzLiteral['plain', 'wrap']zuCallable[[_FieldWrapSerializerT], _FieldWrapSerializerT] | Callable[[_FieldPlainSerializerT], _FieldPlainSerializerT])rJrKrrrGr!cs dddfdd }|S)aDecorator that enables custom field serialization. In the below example, a field of type `set` is used to mitigate duplication. A `field_serializer` is used to serialize the data as a sorted list. ```python from typing import Set from pydantic import BaseModel, field_serializer class StudentModel(BaseModel): name: str = 'Jane' courses: Set[str] @field_serializer('courses', when_used='json') def serialize_courses_in_order(self, courses: Set[str]): return sorted(courses) student = StudentModel(courses={'Math', 'Chemistry', 'English'}) print(student.model_dump_json()) #> {"name":"Jane","courses":["Chemistry","English","Math"]} ``` 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. rC(_decorators.PydanticDescriptorProxy[Any]fr!cs tjd}t||S)N)rJrKrrrG)rZFieldSerializerDecoratorInfoPydanticDescriptorProxyrQZdec_inforGrJrKrrr4r5decszfield_serializer..decr4)rKrrrGrJrUr4rTr5rNs8 ModelPlainSerializerWithInfoModelPlainSerializerWithoutInfoz>ModelPlainSerializerWithInfo | ModelPlainSerializerWithoutInfoModelPlainSerializerModelWrapSerializerWithInfoModelWrapSerializerWithoutInfoz {'unit': 'C', 'value': 100} ``` Two signatures are supported for `mode='plain'`, which is the default: - `(self)` - `(self, info: SerializationInfo)` And two other signatures for `mode='wrap'`: - `(self, nxt: SerializerFunctionWrapHandler)` - `(self, nxt: SerializerFunctionWrapHandler, info: SerializationInfo)` 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. r\rOrPcstjd}t||S)NrKrr)rZModelSerializerDecoratorInforRrSrar4r5rUszmodel_serializer..decNr4)rQrKrrrUr4rar5r_Zs@AnyTypec@s4eZdZdddddZdddddd ZejZd S) SerializeAsAnyr)itemr!cCst|tfSrL)rrc)clsrdr4r4r5__class_getitem__sz SerializeAsAny.__class_getitem__rrrcCsH||}|}|ddkr*|}|d}q tjddtd|d<|S)NtypeZ definitionsr2cSs||SrLr4)xhr4r4r5z=SerializeAsAny.__get_pydantic_core_schema__..)r2r()copyr r?Z any_schema)r1rr r2Zschema_to_updater4r4r5r6s   z+SerializeAsAny.__get_pydantic_core_schema__N)r8r9r:rfr6object__hash__r4r4r4r5rcs rc)N)7r; __future__rZ dataclasses functoolsrrtypingrrrrr r r Z pydantic_corer r Zpydantic_core.core_schemarrrZtyping_extensionsrr _internalrrZannotated_handlersrZ dataclassZ slots_truerr=r@r<rArBrCrDrFrNrVrWrXrYrZr[r\r]r^r_rbrcr4r4r4r5s $   Ef        E      J