
    4gmB                    4   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 ddlmZmZ ddlmZmZmZ ddlmZmZmZ d	d
lmZ d	dlmZmZ d	dlmZ  ej>                  d<i ej@                  ddi G d d             Z! ej>                  d<i ej@                  ddi G d d             Z"er3dZ#de$d<   dZ%de$d<   	 dZ&de$d<   	 dZ'de$d<   	  ede%      Z( ede&      Z)edddd 	 	 	 	 	 	 	 	 	 	 	 	 	 d=d!       Z*eddddd"	 	 	 	 	 	 	 	 	 	 	 	 	 d>d#       Z*d$ed%dd"	 	 	 	 	 	 	 	 	 	 	 d?d&Z*erge
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ge	f   Z.de$d+<   	 e
e	ege	f   Z/de$d,<   	 d-Z0de$d.<   	 d/Z1de$d0<    ed1e-      Z2 ed2e0      Z3ed@d3       Z4ed%dd4	 	 	 	 	 	 	 dAd5       Z4edd%dd6	 	 	 	 	 	 	 dBd7       Z4	 dCd$d%ed6	 	 	 	 	 	 	 	 	 dDd8Z4 ed9      Z5eree5df   Z6y ej>                  d<i ej@                   G d: d;             Z6y)EzEThis module contains related classes and functions for serialization.    )annotationsN)partialpartialmethod)TYPE_CHECKINGAnyCallableTypeVaroverload)PydanticUndefinedcore_schema)SerializationInfoSerializerFunctionWrapHandlerWhenUsed)	AnnotatedLiteral	TypeAlias   )PydanticUndefinedAnnotation)_decorators_internal_dataclass)GetCoreSchemaHandlerfrozenTc                  @    e Zd ZU dZded<   eZded<   dZded<   dd	Zy
)PlainSerializeraC  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 List

    from typing_extensions 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_usedc                    ||      }	 t        j                  | j                  | j                  |j	                         j
                        }|t        u rdn|j                  |      }t        j                  | j                  t        j                  | j                  d      || j                        |d<   |S # t        $ r}t        j                  |      |d}~ww xY w)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   get_function_return_typer   r   _get_types_namespacelocals	NameErrorr   from_name_errorr   generate_schemar   $plain_serializer_function_ser_schemainspect_annotated_serializerr   )selfsource_typehandlerschemar   er&   s          u/var/www/python.lazyprojects.co.uk/rdoDiscordBot/venv/lib/python3.12/site-packages/pydantic/functional_serializers.py__get_pydantic_core_schema__z,PlainSerializer.__get_pydantic_core_schema__7   s     %
	H &>>		  446==K !,/@ @gF]F]^iFj"-"R"RYY ==diiQ'nn	#
   	H-==a@aG	Hs   AB9 9	CCCNr1   r   r2   r   returnzcore_schema.CoreSchema	__name__
__module____qualname____doc____annotations__r   r   r   r6        r5   r   r      s(    > )((K("Ix"r@   r   c                  @    e Zd ZU dZded<   eZded<   dZded<   dd	Zy
)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 Any, Dict

    from typing_extensions import Annotated

    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   c                    ||      }|j                         \  }}	 t        j                  | j                  | j                  |j                         j
                        }|t        u rdn|j                  |      }t        j                  | j                  t        j                  | j                  d      || j                        |d<   |S # t        $ r}t        j                  |      |d}~ww xY w)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,   r   r-   r   #wrap_serializer_function_ser_schemar/   r   )	r0   r1   r2   r3   globalnsr!   r   r4   r&   s	            r5   r6   z+WrapSerializer.__get_pydantic_core_schema__   s     %#88:'
	H &>>		  446==K !,/@ @gF]F]^iFj"-"Q"QYY ==diiP'nn	#
   	H-==a@aG	Hs   AC 	C0C++C0Nr7   r9   r?   r@   r5   rB   rB   W   s)    >@ -,(K("Ix"r@   rB   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_fieldsc                   y Nr?   fieldmoder   r   rN   fieldss         r5   field_serializerrU      s	     @Cr@   )rS   r   r   rN   c                   y rP   r?   rQ   s         r5   rU   rU      s	     BEr@   r"   r   c                $     d f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 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.
    c                d    t        j                        }t        j                  | |      S )N)rT   rS   r   r   rN   )r   FieldSerializerDecoratorInfoPydanticDescriptorProxy)fdec_inforN   rT   rS   r   r   s     r5   deczfield_serializer.<locals>.dec  s7    ;;#%
 221h??r@   )r[   rJ   r8   (_decorators.PydanticDescriptorProxy[Any]r?   )rS   r   r   rN   rT   r]   s   ````` r5   rU   rU      s    p@ @ Jr@   ModelPlainSerializerWithInfoModelPlainSerializerWithoutInfoz>ModelPlainSerializerWithInfo | ModelPlainSerializerWithoutInfoModelPlainSerializerModelWrapSerializerWithInfoModelWrapSerializerWithoutInfoz<ModelWrapSerializerWithInfo | ModelWrapSerializerWithoutInfoModelWrapSerializerz*ModelPlainSerializer | ModelWrapSerializerModelSerializer_ModelPlainSerializerT_ModelWrapSerializerTc                    y rP   r?   )r[   s    r5   model_serializerri   F  s    NQr@   )r   r   c                     y rP   r?   rS   r   r   s      r5   ri   ri   J  s	     @Cr@   rk   c                     y rP   r?   rk   s      r5   ri   ri   P  s	     BEr@   c              0    dfd}| |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 [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.
    c                `    t        j                        }t        j                  | |      S )N)rS   r   r   )r   ModelSerializerDecoratorInforZ   )r[   r\   rS   r   r   s     r5   r]   zmodel_serializer.<locals>.dec  s,    ;;S^jst221h??r@   )r[   re   r8   r^   r?   )r[   rS   r   r   r]   s    ``` r5   ri   ri   Y  s    @@ 	y
1vr@   AnyTypec                  @    e Zd ZddZ	 	 	 	 	 	 ddZej                  Zy)SerializeAsAnyc                (    t         |t               f   S rP   )r   rr   )clsitems     r5   __class_getitem__z SerializeAsAny.__class_getitem__  s    T>#3344r@   c                     ||      }|}|d   dk(  r|j                         }|d   }|d   dk(  rt        j                  d t        j                               |d<   |S )Ntypedefinitionsr3   c                     ||       S rP   r?   )xhs     r5   <lambda>z=SerializeAsAny.__get_pydantic_core_schema__.<locals>.<lambda>  s    QqTr@   )r3   r'   )copyr   rE   
any_schema)r0   r1   r2   r3   schema_to_updates        r5   r6   z+SerializeAsAny.__get_pydantic_core_schema__  sw     [)F%"6*m;#3#8#8#: #3H#=  #6*m; 1<0_0_!+*@*@*B1_- Mr@   N)ru   r   r8   r   r7   )r:   r;   r<   rv   r6   object__hash__r?   r@   r5   rr   rr     s/    	5	"	-A	#	 ??r@   rr   r?   )rR   strrT   r   rS   Literal['wrap']r   r   r   r   rN   bool | Noner8   z8Callable[[_FieldWrapSerializerT], _FieldWrapSerializerT])rR   r   rT   r   rS   Literal['plain']r   r   r   r   rN   r   r8   z:Callable[[_FieldPlainSerializerT], _FieldPlainSerializerT])rT   r   rS   Literal['plain', 'wrap']r   r   r   r   rN   r   r8   zuCallable[[_FieldWrapSerializerT], _FieldWrapSerializerT] | Callable[[_FieldPlainSerializerT], _FieldPlainSerializerT])r[   rf   r8   rf   )rS   r   r   r   r   r   r8   z8Callable[[_ModelWrapSerializerT], _ModelWrapSerializerT])rS   r   r   r   r   r   r8   z:Callable[[_ModelPlainSerializerT], _ModelPlainSerializerT]rP   )
r[   z5_ModelPlainSerializerT | _ModelWrapSerializerT | NonerS   r   r   r   r   r   r8   z_ModelPlainSerializerT | Callable[[_ModelWrapSerializerT], _ModelWrapSerializerT] | Callable[[_ModelPlainSerializerT], _ModelPlainSerializerT])7r=   
__future__r   dataclasses	functoolsr   r   typingr   r   r   r	   r
   pydantic_corer   r   pydantic_core.core_schemar   r   r   typing_extensionsr   r   r    r   	_internalr   r   annotated_handlersr   	dataclass
slots_truer   rB   rG   r>   rH   rI   rJ   rK   rM   rU   r_   r`   ra   rb   rc   rd   re   rf   rg   ri   rp   rr   r?   r@   r5   <module>r      s2   K "  , B B 8 ` ` ; ; ) 7 4 E,77EEA A FAH E,77EEc c FcL =Hi=&Q)Q@%TT?!MOYM0$%=EYZ#$;CVW 
  #CC C 	C
 C C C >C 
C 

 ! #EE E 	E
 E E E @E 
E &-(" $BB
"B B 	B
 BABJ  /7=N7OQT7T.U )UN193%*1E#YEQ&f)f4-5s<Y[l6mor6r-ssM08#?\9]_b9b0c"IcP%cc3!MOYM$%=EYZ#$;CVW 
 Q 
 Q 
4<QTCC)1CKNC=C 
C
 
 !"	E
E E 	E
 @E 
E @DG &-"(G<G #	G
 G GAGT )
 w|,N [<0;;<# # =#r@   