o
    niF                     @  s  U d Z ddlmZ ddlZddlmZmZ ddlmZm	Z	m
Z
mZmZmZmZ ddlmZmZ ddlmZmZmZ ddlmZ d	d
lmZ d	dlmZmZ d	dlmZ d	dlm Z  ej!dUi ej"ddiG dd dZ#ej!dUi ej"ddiG dd dZ$erdZ%de&d< dZ'de&d< 	 dZ(de&d< 	 dZ)de&d< 	 ede'dZ*ede(dZ+ed d d d!dVd/d0Z,ed d d d d1dWd4d0Z,d5ed6dd1dXd9d0Z,er)ee
ee
 ge
f Z-de&d:< 	 ee
ge
f Z.de&d;< 	 d<Z/de&d=< 	 ee
eee
 ge
f Z0de&d>< 	 ee
ege
f Z1de&d?< 	 d@Z2de&dA< 	 dBZ3de&dC< edDe/dZ4edEe2dZ5edYdGdHZ6ed6d dIdZdKdHZ6ed d6d dLd[dNdHZ6	d\d5d6edLd]dQdHZ6edRZ7erbe	e7d f Z8dS ej!dUi ej"G dSdT dTZ8dS )^zEThis module contains related classes and functions for serialization.    )annotationsN)partialpartialmethod)TYPE_CHECKING	AnnotatedAnyCallableLiteralTypeVaroverload)PydanticUndefinedcore_schema)SerializationInfoSerializerFunctionWrapHandlerWhenUsed)	TypeAlias   )PydanticUndefinedAnnotation)_decorators_internal_dataclass)GetCoreSchemaHandler)PydanticUserErrorfrozenTc                   @  <   e Zd ZU dZded< eZded< dZded< dddZdS )PlainSerializera  Plain 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'.
    zcore_schema.SerializerFunctionfuncr   return_typealwaysr   	when_usedsource_typehandlerr   returncore_schema.CoreSchemac              
   C     ||}| j tur| j }n ztj| j| jd}W n ty, } zt	||d}~ww |tu r3dn|
|}tj| jt| jd|| jd|d< |S )zGets the Pydantic core schema.

        Args:
            source_type: The source type.
            handler: The `GetCoreSchemaHandler` instance.

        Returns:
            The Pydantic core schema.
        localnsNplainfunctioninfo_argreturn_schemar   serialization)r   r   r   get_callable_return_typer   _get_types_namespacelocals	NameErrorr   from_name_errorgenerate_schemar   $plain_serializer_function_ser_schemainspect_annotated_serializerr   selfr   r    schemar   er*    r8   _/var/www/html/arapca_proje/venv/lib/python3.10/site-packages/pydantic/functional_serializers.py__get_pydantic_core_schema__6   (   



z,PlainSerializer.__get_pydantic_core_schema__Nr   r   r    r   r!   r"   	__name__
__module____qualname____doc____annotations__r   r   r   r:   r8   r8   r8   r9   r      s   
 r   c                   @  r   )WrapSerializera  Wrap 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.WrapSerializerFunctionr   r   r   r   r   r   r   r    r   r!   r"   c              
   C  r#   )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$   Nwrapr'   r+   )r   r   r   r,   r   r-   r.   r/   r   r0   r1   r   #wrap_serializer_function_ser_schemar3   r   r4   r8   r8   r9   r:      r;   z+WrapSerializer.__get_pydantic_core_schema__Nr<   r=   r8   r8   r8   r9   rC   Y   s   
 >rC   z!partial[Any] | partialmethod[Any]r   _Partialz)core_schema.SerializerFunction | _PartialFieldPlainSerializerz-core_schema.WrapSerializerFunction | _PartialFieldWrapSerializerz*FieldPlainSerializer | FieldWrapSerializerFieldSerializer_FieldPlainSerializerT)bound_FieldWrapSerializerT.)r   r   check_fieldsfieldstrfieldsmodeLiteral['wrap']r   r   r   r   rM   bool | Noner!   8Callable[[_FieldWrapSerializerT], _FieldWrapSerializerT]c               G     d S Nr8   rN   rQ   r   r   rM   rP   r8   r8   r9   field_serializer      	rX   )rQ   r   r   rM   Literal['plain']:Callable[[_FieldPlainSerializerT], _FieldPlainSerializerT]c               G  rU   rV   r8   rW   r8   r8   r9   rX      rY   r&   r   Literal['plain', 'wrap']uCallable[[_FieldWrapSerializerT], _FieldWrapSerializerT] | Callable[[_FieldPlainSerializerT], _FieldPlainSerializerT]c                 sb   t | s	t| trtddd| gR tdd D s$tdddd fdd}|S )a  Decorator 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 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 [the usage documentation](../concepts/serialization.md#serializers) for more information.

    Four signatures are supported for the decorated serializer:

    - `(self, value: Any, info: FieldSerializationInfo)`
    - `(self, value: Any, nxt: SerializerFunctionWrapHandler, info: FieldSerializationInfo)`
    - `(value: Any, info: SerializationInfo)`
    - `(value: Any, nxt: SerializerFunctionWrapHandler, info: SerializationInfo)`

    Args:
        *fields: The field names the serializer should apply to.
        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.

    Raises:
        PydanticUserError:
            - If the decorator is used without any arguments (at least one field name must be provided).
            - If the provided field names are not strings.
    zThe `@field_serializer` decorator cannot be used without arguments, at least one field must be provided. For example: `@field_serializer('<field_name>', ...)`.zdecorator-missing-arguments)codec                 s  s    | ]}t |tV  qd S rV   )
isinstancerO   ).0rN   r8   r8   r9   	<genexpr>*  s    z#field_serializer.<locals>.<genexpr>zThe provided field names to the `@field_serializer` decorator should be strings. For example: `@field_serializer('<field_name_1>', '<field_name_2>', ...).`zdecorator-invalid-fieldsfrI   r!   (_decorators.PydanticDescriptorProxy[Any]c                   s    t j d}t | |S )N)rP   rQ   r   r   rM   )r   FieldSerializerDecoratorInfoPydanticDescriptorProxyrb   dec_inforM   rP   rQ   r   r   r8   r9   dec1  s   zfield_serializer.<locals>.decN)rb   rI   r!   rc   )callabler_   classmethodr   all)rN   rQ   r   r   rM   rP   ri   r8   rh   r9   rX      s   :
ModelPlainSerializerWithInfoModelPlainSerializerWithoutInfoz>ModelPlainSerializerWithInfo | ModelPlainSerializerWithoutInfoModelPlainSerializerModelWrapSerializerWithInfoModelWrapSerializerWithoutInfoz<ModelWrapSerializerWithInfo | ModelWrapSerializerWithoutInfoModelWrapSerializerz*ModelPlainSerializer | ModelWrapSerializerModelSerializer_ModelPlainSerializerT_ModelWrapSerializerTrb   c                C  rU   rV   r8   )rb   r8   r8   r9   model_serializerY  s   rv   )r   r   8Callable[[_ModelWrapSerializerT], _ModelWrapSerializerT]c                 C  rU   rV   r8   rQ   r   r   r8   r8   r9   rv   ]  s   rx   :Callable[[_ModelPlainSerializerT], _ModelPlainSerializerT]c                 C  rU   rV   r8   rx   r8   r8   r9   rv   c  s   5_ModelPlainSerializerT | _ModelWrapSerializerT | None_ModelPlainSerializerT | Callable[[_ModelWrapSerializerT], _ModelWrapSerializerT] | Callable[[_ModelPlainSerializerT], _ModelPlainSerializerT]c                 s&   d fdd}| du r|S || S )	a!  Decorator that enables custom model serialization.

    This is useful when a model need to be serialized in a customized manner, allowing for flexibility beyond just specific fields.

    An example would be to serialize temperature to the same temperature scale, such as degrees Celsius.

    ```python
    from typing import Literal

    from pydantic import BaseModel, model_serializer

    class TemperatureModel(BaseModel):
        unit: Literal['C', 'F']
        value: int

        @model_serializer()
        def serialize_model(self):
            if self.unit == 'F':
                return {'unit': 'C', 'value': int((self.value - 32) / 1.8)}
            return {'unit': self.unit, 'value': self.value}

    temperature = TemperatureModel(unit='F', value=212)
    print(temperature.model_dump())
    #> {'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 [the usage documentation](../concepts/serialization.md#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.
    rb   rs   r!   rc   c                   s   t j d}t | |S )NrQ   r   r   )r   ModelSerializerDecoratorInfore   rf   r|   r8   r9   ri     s   zmodel_serializer.<locals>.decN)rb   rs   r!   rc   r8   )rb   rQ   r   r   ri   r8   r|   r9   rv   l  s   @AnyTypec                   @  s*   e Zd ZdZdddZdddZejZdS )SerializeAsAnyzAnnotation used to mark a type as having duck-typing serialization behavior.

        See [usage documentation](../concepts/serialization.md#serializing-with-duck-typing) for more details.
        itemr   r!   c                 C  s   t |t f S rV   )r   r   )clsr   r8   r8   r9   __class_getitem__  s   z SerializeAsAny.__class_getitem__r   r    r   r"   c                 C  sF   ||}|}|d dkr|  }|d }|d dkstd|d< |S )Ntypedefinitionsr6   anyr+   )copyr   simple_ser_schema)r5   r   r    r6   schema_to_updater8   r8   r9   r:     s   z+SerializeAsAny.__get_pydantic_core_schema__N)r   r   r!   r   r<   )r>   r?   r@   rA   r   r:   object__hash__r8   r8   r8   r9   r     s
    


r   r8   )rN   rO   rP   rO   rQ   rR   r   r   r   r   rM   rS   r!   rT   )rN   rO   rP   rO   rQ   rZ   r   r   r   r   rM   rS   r!   r[   )rN   rO   rP   rO   rQ   r\   r   r   r   r   rM   rS   r!   r]   )rb   rt   r!   rt   )rQ   rR   r   r   r   r   r!   rw   )rQ   rZ   r   r   r   r   r!   ry   rV   )
rb   rz   rQ   r\   r   r   r   r   r!   r{   )9rA   
__future__r   dataclasses	functoolsr   r   typingr   r   r   r   r	   r
   r   pydantic_corer   r   pydantic_core.core_schemar   r   r   typing_extensionsr    r   	_internalr   r   annotated_handlersr   errorsr   	dataclass
slots_truer   rC   rF   rB   rG   rH   rI   rJ   rL   rX   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   r~   r   r8   r8   r8   r9   <module>   s    $EfV	J