o
    ZhW?                     @  sV  d Z ddlmZ ddlZddlZddlZddlmZm	Z	m
Z
mZmZmZmZmZ ddlmZ ddlmZmZ ddlmZmZmZmZ dd	lmZ dd
lmZ ddlmZ ddl m!Z! ddl"m#Z#m$Z$m%Z% errddl&m'Z' ddl(m)Z) dZ*edZ+ej,dkreej-e#e%fdeddddddddddd
dKd(d)Z.eej-e#e%fdeddddddddddd
dLd-d)Z.n4eej-e#e%fdeddddddddd.dMd/d)Z.eej-e#e%fdeddddddddd.dNd0d)Z.eej-e#e%fd	dOddddddddddd
dPd3d)Z.ee/Z0ej,d4k rdQd9d:Z1e1ej2_3ddd;dd<dRdDdEZ4dSdIdJZ5dS )Tz7Provide an enhanced dataclass that performs validation.    )annotationsN)TYPE_CHECKINGAnyCallableGenericLiteralNoReturnTypeVaroverload)warn)	TypeGuarddataclass_transform   )_config_decorators_namespace_utils_typing_extra)_dataclasses)getattr_migration)
ConfigDict)PydanticUserError)Field	FieldInfoPrivateAttr)PydanticDataclass)MappingNamespace)	dataclassrebuild_dataclass_T   
   )field_specifiersFT.
initrepreqorderunsafe_hashfrozenconfigvalidate_on_initkw_onlyslotsr$   Literal[False]r%   boolr&   r'   r(   r)   r*    ConfigDict | type[object] | Noner+   bool | Noner,   r-   return-Callable[[type[_T]], type[PydanticDataclass]]c        
   
      C     d S N r#   r6   r6   K/var/www/html/lang_env/lib/python3.10/site-packages/pydantic/dataclasses.pyr         r   _clstype[_T]type[PydanticDataclass]c       
         C  r4   r5   r6   )r9   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r6   r6   r7   r   /   r8   r$   r%   r&   r'   r(   r)   r*   r+   c                 C  r4   r5   r6   r<   r6   r6   r7   r   B      c          	      C  r4   r5   r6   )	r9   r$   r%   r&   r'   r(   r)   r*   r+   r6   r6   r7   r   Q   r=   type[_T] | NoneGCallable[[type[_T]], type[PydanticDataclass]] | type[PydanticDataclass]c       
      	     st   |du sJ d|dusJ dt jdkr|	|
dni dd
dd fdd}| du r6|S || S )a  !!! abstract "Usage Documentation"
        [`dataclasses`](../concepts/dataclasses.md)

    A decorator used to create a Pydantic-enhanced dataclass, similar to the standard Python `dataclass`,
    but with added validation.

    This function should be used similarly to `dataclasses.dataclass`.

    Args:
        _cls: The target `dataclass`.
        init: Included for signature compatibility with `dataclasses.dataclass`, and is passed through to
            `dataclasses.dataclass` when appropriate. If specified, must be set to `False`, as pydantic inserts its
            own  `__init__` function.
        repr: A boolean indicating whether to include the field in the `__repr__` output.
        eq: Determines if a `__eq__` method should be generated for the class.
        order: Determines if comparison magic methods should be generated, such as `__lt__`, but not `__eq__`.
        unsafe_hash: Determines if a `__hash__` method should be included in the class, as in `dataclasses.dataclass`.
        frozen: Determines if the generated class should be a 'frozen' `dataclass`, which does not allow its
            attributes to be modified after it has been initialized. If not set, the value from the provided `config` argument will be used (and will default to `False` otherwise).
        config: The Pydantic config to use for the `dataclass`.
        validate_on_init: A deprecated parameter included for backwards compatibility; in V2, all Pydantic dataclasses
            are validated on init.
        kw_only: Determines if `__init__` method parameters must be specified by keyword only. Defaults to `False`.
        slots: Determines if the generated class should be a 'slots' `dataclass`, which does not allow the addition of
            new attributes after instantiation.

    Returns:
        A decorator that accepts a class as its argument and returns a Pydantic `dataclass`.

    Raises:
        AssertionError: Raised if `init` is not `False` or `validate_on_init` is `False`.
    Fz7pydantic.dataclasses.dataclass only supports init=Falsez-validate_on_init=False is no longer supportedr   )r,   r-   cls	type[Any]r2   Nonec              	   S  s   | j D ]Q}t|di }|D ]F}t| |d}t|tsqd|i}tjdkr+|jr+d|d< |jdur5|j|d< t| |t	j
di | | jddu rLi | _|| | j|< qqdS )	a  Make sure that stdlib `dataclasses` understands `Field` kwargs like `kw_only`
        To do that, we simply change
          `x: int = pydantic.Field(..., kw_only=True)`
        into
          `x: int = dataclasses.field(default=pydantic.Field(..., kw_only=True), kw_only=True)`
        __annotations__Ndefaultr   Tr,   r%   r6   )__mro__getattr
isinstancer   sysversion_infor,   r%   setattrdataclassesfield__dict__getrC   )r@   Zannotation_clsr   
field_nameZfield_valueZ
field_argsr6   r6   r7   make_pydantic_fields_compatible   s"   



z2dataclass.<locals>.make_pydantic_fields_compatibler;   c              	     s~  ddl m} || rtd| j ddd| }tdd | jD }|s8 d	ur8t| d
r8td| j dtdd  d	ur> nt	| d
d	}t
|}tj| }| j}t| rrd	}| f}t| trkt| j }	||	f }t| j|} |  d	ur}
|jrtd| jdtdd n|jpd}
tj| fd|
d} d| _|| _|| _|j| _|j| _d| _tj| |dd | S )zCreate a Pydantic dataclass from a regular dataclass.

        Args:
            cls: The class to create the Pydantic dataclass from.

        Returns:
            A Pydantic dataclass.
        r   )is_model_classz(Cannot create a Pydantic dataclass from z" as it is already a Pydantic modelzdataclass-on-model)codec                 s  s    | ]}t |V  qd S r5   )rK   is_dataclass).0baser6   r6   r7   	<genexpr>   s    z6dataclass.<locals>.create_dataclass.<locals>.<genexpr>N__pydantic_config__z[`config` is set via both the `dataclass` decorator and `__pydantic_config__` for dataclass zK. The `config` specification from `dataclass` decorator will take priority.   )category
stacklevelzN`frozen` is set via both the `dataclass` decorator and `config` for dataclass zW.This is not recommended. The `frozen` specification on `dataclass` will take priority.FT)r$   r%   r&   r'   r(   r)   )raise_errors) Z_internal._utilsrQ   r   __name__any	__bases__hasattrr   UserWarningrF   r   ConfigWrapperr   ZDecoratorInfosbuild__doc___pydantic_dataclassesZis_builtin_dataclass
issubclassr   __parameters__types	new_classr)   rK   r   __is_pydantic_dataclass__Z__pydantic_decorators__
__module____qualname____pydantic_complete__complete_dataclass)r@   rQ   Zoriginal_clsZhas_dataclass_baseZconfig_dictZconfig_wrapperZ
decoratorsZoriginal_docbasesgeneric_baseZfrozen_r*   r&   r)   kwargsrP   r'   r%   r(   r6   r7   create_dataclass   sp   	





	z#dataclass.<locals>.create_dataclassN)r@   rA   r2   rB   )r@   rA   r2   r;   )rH   rI   )r9   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   rr   r6   rp   r7   r   a   s   /

!a)r       argsr   rq   r   c                  O  s   t d)a9  This function does nothing but raise an error that is as similar as possible to what you'd get
        if you were to try calling `InitVar[int]()` without this monkeypatch. The whole purpose is just
        to ensure typing._type_check does not error if the type hint evaluates to `InitVar[<parameter>]`.
        z 'InitVar' object is not callable)	TypeError)rt   rq   r6   r6   r7   _call_initvar#  s   rv   rX   )forcer[   _parent_namespace_depth_types_namespacer@   rw   r[   rx   intry   MappingNamespace | Nonec                C  s   |s| j rdS dD ]}|| jv rt| | q	d| _ |dur |}n|dkr.tj|ddp,i }ni }tj|d}tj| t	j
| jdd||dd	S )
ax  Try to rebuild the pydantic-core schema for the dataclass.

    This may be necessary when one of the annotations is a ForwardRef which could not be resolved during
    the initial attempt to build the schema, and automatic rebuilding fails.

    This is analogous to `BaseModel.model_rebuild`.

    Args:
        cls: The class to rebuild the pydantic-core schema for.
        force: Whether to force the rebuilding of the schema, defaults to `False`.
        raise_errors: Whether to raise errors, defaults to `True`.
        _parent_namespace_depth: The depth level of the parent namespace, defaults to 2.
        _types_namespace: The types namespace, defaults to `None`.

    Returns:
        Returns `None` if the schema is already "complete" and rebuilding was not required.
        If rebuilding _was_ required, returns `True` if rebuilding was successful, otherwise `False`.
    N)Z__pydantic_core_schema__Z__pydantic_validator__Z__pydantic_serializer__Fr   T)Zparent_depthrw   )Zparent_namespace)check)r[   ns_resolverZ_force_build)rl   rM   delattrr   Zparent_frame_namespacer   Z
NsResolverrd   rm   r   ra   rW   )r@   rw   r[   rx   ry   attrZ
rebuild_nsr}   r6   r6   r7   r   -  s,   


r   class_rA   "TypeGuard[type[PydanticDataclass]]c                C  s,   zd| j v o
t| W S  ty   Y dS w )zWhether a class is a pydantic dataclass.

    Args:
        class_: The class.

    Returns:
        `True` if the class is a pydantic dataclass, `False` otherwise.
    ri   F)rM   rK   rS   AttributeError)r   r6   r6   r7   is_pydantic_dataclassj  s
   	r   )r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r/   r*   r0   r+   r1   r,   r/   r-   r/   r2   r3   )r9   r:   r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r,   r/   r-   r/   r2   r;   )r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r2   r3   )r9   r:   r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r2   r;   r5   )r9   r>   r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r,   r/   r-   r/   r2   r?   )rt   r   rq   r   r2   r   )r@   r;   rw   r/   r[   r/   rx   rz   ry   r{   r2   r1   )r   rA   r2   r   )6rc   
__future__r   Z_annotationsrK   rH   rg   typingr   r   r   r   r   r   r	   r
   warningsr   typing_extensionsr   r   	_internalr   r   r   r   r   rd   Z
_migrationr   r*   r   errorsr   fieldsr   r   r   Z_internal._dataclassesr   Z_internal._namespace_utilsr   __all__r   rI   rL   r   r\   __getattr__rv   InitVar__call__r   r   r6   r6   r6   r7   <module>   s    (
 <
=