o
    g                     @  s  d Z ddlmZ ddlZddlZddlZddlZddlZddl	Z	ddl
mZ ddlmZmZ ddlmZ ddl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mZ d
dlm Z m!Z! d
dl"m#Z# d
dlm$Z$m%Z% ddl&m'Z' ddl(m)Z)m*Z*m+Z+m,Z, ddl-m.Z.m/Z/m0Z0 ddl1m2Z2 ddl3m4Z4m5Z5 ddl6m7Z7m8Z8 ddl9m:Z: ddl;m<Z< ddl=m>Z> ddl?m@Z@ ddlAmBZBmCZCmDZDmEZEmFZF ddlGmHZHmIZI ejJrd
dlKmLZLmMZMmNZN d
dlKmOZP d
dlKmQZR d
dlSmTZT ne%ZUeV ZPeV ZReVjWZXG dd  d eYZZd!d"ded'd(Z[ed)ePeRe[fd*G d+d, d,eZ\dfd1d2Z]dgd8d9Z^dhd?d@Z_didCdDZ`djdEdFZadkdKdLZbd)dddMdldTdUZcdmdVdWZdG dXdY dYZeG dZd[ d[Zfdnd^d_Zgdnd`daZheddbdodcddZidS )pz"Private logic for creating models.    )annotationsN)ABCMeta)	lru_cachepartial)FunctionType)AnyCallableGenericLiteralNoReturncast)PydanticUndefinedSchemaSerializer)TypeAliasTypedataclass_transform
deprecatedget_args   )PydanticUndefinedAnnotationPydanticUserError)create_schema_validator)GenericBeforeBaseModelWarningPydanticDeprecatedSince20   )ConfigWrapper)DecoratorInfosPydanticDescriptorProxyget_attribute_from_basesunwrap_wrapped_function)collect_model_fieldsis_valid_field_nameis_valid_privateattr_name)GenerateSchema)PydanticGenericMetadataget_model_typevars_map)import_cached_base_modelimport_cached_field_info)set_model_mocks)
NsResolver)CallbackGetCoreSchemaHandler)generate_pydantic_signature)_make_forward_refeval_type_backportis_annotatedis_classvar_annotationparent_frame_namespace)LazyClassAttributeSafeGetItemProxy)ComputedFieldInfo	FieldInfoModelPrivateAttr)Field)PrivateAttr)	BaseModelc                      s"   e Zd ZdZd
 fdd	Z  ZS )_ModelNamespaceDictz{A dictionary subclass that intercepts attribute setting on model classes and
    warns about overriding of decorators.
    kstrvobjectreturnNonec                   sL   |  |d }|r||urt|trtd| d|jj d t ||S )N`z"` overrides an existing Pydantic `z` decorator)	get
isinstancer   warningswarndecorator_infodecorator_reprsuper__setitem__)selfr9   r;   existing	__class__ ]/var/www/visachat/venv/lib/python3.10/site-packages/pydantic/_internal/_model_construction.pyrG   =   s   z_ModelNamespaceDict.__setitem__)r9   r:   r;   r<   r=   r>   )__name__
__module____qualname____doc__rG   __classcell__rL   rL   rJ   rM   r8   8   s    r8   FinitrT   Literal[False]r=   r   c                 C     dS )zOnly for typing purposes. Used as default value of `__pydantic_fields_set__`,
    `__pydantic_extra__`, `__pydantic_private__`, so they could be ignored when
    synthesizing the `__init__` signature.
    NrL   rS   rL   rL   rM   NoInitFieldE       rW   T)kw_only_defaultfield_specifiersc                      s   e Zd Z			d/d0 fddZejsd1ddZed2ddZd3 fddZ	e
d4d d!Zeed"dd#d5d%d&Zed5d'd(Zed6d*d+Zd7 fd-d.Z  ZS )8ModelMetaclassNTcls_namer:   basestuple[type[Any], ...]	namespacedict[str, Any]__pydantic_generic_metadata__PydanticGenericMetadata | None#__pydantic_reset_parent_namespace__bool_create_model_module
str | Nonekwargsr   r=   typec                   s  |r|  |\}}	}
t|||}|j|d< t||j|	|}|s#|
r<t||  dur8d5 fd	d
}||d< nt|d< |	|d< i |
||d< tdt	 j
| |||fi |}t }|j}t|v ru|t||k rutjtddd t|jdd |_|j|ju rdnd|_t||_|r||_nt|di ddt|ddpr,r,tfddD s,ddlm} tfddD }|v r|vrd dd |D }|j! d| d |j! d!| d"}t"|| }d d#d |D }d$| d%}d&| d'}t|vr(d d(d |D |g }|d)|j! d*| d+7 }t"|ddd,|_d|_#|$ D ]\}}|%|| q:|rOt&t' |_(t|d-d}t)|t*r_t+|}t,|d.}t-|||| |j.ryd/|vryt/|| t0|||d||d0 d1d2 |jj1$ D |_2t3| t	||j4di | |S d3D ]	}|5|d q|d4i 6  t	 j
| |||fi |S )6a  Metaclass for creating Pydantic models.

        Args:
            cls_name: The name of the class to be created.
            bases: The base classes of the class to be created.
            namespace: The attribute dictionary of the class to be created.
            __pydantic_generic_metadata__: Metadata for generic models.
            __pydantic_reset_parent_namespace__: Reset parent namespace.
            _create_model_module: The module of the class to be created, if created by `create_model`.
            **kwargs: Catch-all for any other keyword arguments.

        Returns:
            The new class created by the metaclass.
        model_configNrH   r7   contextr   r=   r>   c                  s   t | |  | | dS )zWe need to both initialize private attributes and call the user-defined model_post_init
                        method.
                        N)init_private_attributes)rH   rj   )original_model_post_initrL   rM   wrapped_model_post_inity   s   
z7ModelMetaclass.__new__.<locals>.wrapped_model_post_initmodel_post_init__class_vars____private_attributes__type[BaseModel]zClasses should inherit from `BaseModel` before generic classes (e.g. `typing.Generic[T]`) for pydantic generics to work properly.r   
stacklevel__pydantic_base_init__Fra   
parametersrL   __parameters__c                 3  s    | ]}| v V  qd S NrL   .0x)ru   rL   rM   	<genexpr>   s    z)ModelMetaclass.__new__.<locals>.<genexpr>)RootModelRootTypec                 3  s    | ]	}| vr|V  qd S rw   rL   rx   )parent_parametersrL   rM   r{      s    z, c                 S     g | ]}|j qS rL   rN   rx   rL   rL   rM   
<listcomp>       z*ModelMetaclass.__new__.<locals>.<listcomp>zS is a subclass of `RootModel`, but does not include the generic type identifier(s) zL in its parameters. You should parametrize RootModel directly, e.g., `class z(RootModel[z	]): ...`.c                 S  s   g | ]}t |qS rL   )r:   rx   rL   rL   rM   r      s    ztyping.Generic[]zJAll parameters must be present on typing.Generic; you should inherit from .c                 S  r~   rL   r   rx   rL   rL   rM   r      r   z- Note: `typing.Generic` must go last: `class (z): ...`))originargsru   __pydantic_parent_namespace__)parent_namespace__hash__raise_errorsns_resolvercreate_model_modulec                 S  s   i | ]\}}||j qS rL   )info)ry   r9   r;   rL   rL   rM   
<dictcomp>   s    z*ModelMetaclass.__new__.<locals>.<dictcomp>)__pydantic_fields_set____pydantic_extra____pydantic_private____annotations__rH   r7   rj   r   r=   r>   )7_collect_bases_datar   	for_modelconfig_dictinspect_namespaceignored_typesget_model_post_initrk   r   rF   __new__r%   __mro__r	   indexrB   rC   r   getattr__init____pydantic_custom_init__rn   __pydantic_post_init__r   build__pydantic_decorators__ra   r@   all
root_modelr|   tuplejoinrN   	TypeError__pydantic_complete__items__set_name__build_lenient_weakvaluedictr/   r   rA   dictunpack_lenient_weakvaluedictr(   set_model_fieldsfrozenset_default_hash_funccomplete_model_classcomputed_fields__pydantic_computed_fields__set_deprecated_descriptors__pydantic_init_subclass__popclear)mcsr\   r]   r_   ra   rc   re   rg   base_field_names
class_varsbase_private_attributesconfig_wrapperprivate_attributesrm   cls
BaseModel_mror|   missing_parametersparameters_strerror_messagecombined_parametersgeneric_type_label	bases_strnameobjr   r   instance_slotrJ   )rl   ru   r}   rM   r   Q   s   



 $



zModelMetaclass.__new__itemc                 C  s(   | j d}|r||v r|| S t|)zNThis is necessary to keep attribute access working for class attribute access.rp   )__dict__r@   AttributeError)rH   r   r   rL   rL   rM   __getattr__  s   zModelMetaclass.__getattr__r   dict[str, object]c                 O  s   t  S rw   )r8   )r   r   rg   rL   rL   rM   __prepare__
  s   zModelMetaclass.__prepare__instancec                   s   t |do
t |S )zsAvoid calling ABC _abc_subclasscheck unless we're pretty sure.

        See #3829 and python/cpython#92810
        __pydantic_validator__)hasattrrF   __instancecheck__)rH   r   rJ   rL   rM   r     s   z ModelMetaclass.__instancecheck__6tuple[set[str], set[str], dict[str, ModelPrivateAttr]]c                 C  sj   t  }t }t }i }| D ]"}t||r/||ur/|t|di   ||j ||j q|||fS )N__pydantic_fields__)r%   set
issubclassupdater   keysro   rp   )r]   r7   field_namesr   r   baserL   rL   rM   r     s   
z"ModelMetaclass._collect_bases_dataEThe `__fields__` attribute is deprecated, use `model_fields` instead.)categorydict[str, FieldInfo]c                 C  s   t jdtdd | jS )Nr   r   rr   )rB   rC   r   model_fieldsrH   rL   rL   rM   
__fields__$  s   zModelMetaclass.__fields__c                 C     t | di S )zGet metadata about the fields defined on the model.

        Returns:
            A mapping of field names to [`FieldInfo`][pydantic.fields.FieldInfo] objects.
        r   r   r   rL   rL   rM   r   .     zModelMetaclass.model_fieldsdict[str, ComputedFieldInfo]c                 C  r   )zGet metadata about the computed fields defined on the model.

        Returns:
            A mapping of computed field names to [`ComputedFieldInfo`][pydantic.fields.ComputedFieldInfo] objects.
        r   r   r   rL   rL   rM   model_computed_fields7  r   z$ModelMetaclass.model_computed_fields	list[str]c                   s$   t t  }d|v r|d |S )Nr   )listrF   __dir__remove)rH   
attributesrJ   rL   rM   r   @  s   
zModelMetaclass.__dir__)NTN)r\   r:   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   r   r=   rh   )r   r:   r=   r   )r   r   rg   r   r=   r   )r   r   r=   rd   )r]   r^   r=   r   )r=   r   )r=   r   )r=   r   )rN   rO   rP   r   typingTYPE_CHECKINGr   classmethodr   r   staticmethodr   propertyr   r   r   r   r   rR   rL   rL   rJ   rM   r[   O   s*     0

r[   rH   r7   rj   r>   c                C  sT   t | dddu r(i }| j D ]\}}| }|tur|||< qt| d| dS dS )a  This function is meant to behave like a BaseModel method to initialise private attributes.

    It takes context as an argument since that's what pydantic-core passes when calling it.

    Args:
        self: The BaseModel instance.
        context: The context.
    r   N)r   rp   r   get_defaultr   object_setattr)rH   rj   pydantic_privater   private_attrdefaultrL   rL   rM   rk   G  s   	rk   r_   r`   r]   r^   Callable[..., Any] | Nonec                 C  s2   d| v r| d S t  }t|d}||jur|S dS )zaGet the `model_post_init` method from the namespace or the class bases, or `None` if not defined.rn   N)r%   r   rn   )r_   r]   r7   rn   rL   rL   rM   r   Y  s   

r   r   base_class_varsset[str]base_class_fieldsdict[str, ModelPrivateAttr]c              
     s  ddl m m} t }|t  }i }| di }d|v s d| v r$tdt }	t| 	 D ]\}
}|
dks9|
dkr:q-t
|trS|j| d krSd	| v rS|j| d	 rSq-t
||s^|jjd
krd|	|
 q-t
| r|
drvtd|
dt|
rtdd|
 d|
d|||
< | |
= q-t
||rt|
s|
dpd}td|d|
d|
drq-t|
r|
|vst||
 st ||d||
< | |
= q-|
|v rq-|
|vr|
|v rtd|
dddt
||rtd|
dddtd|
 d|d|
 dddq-|	 D ]{\}}t|r||vr||	vrt|s||vrt|ddd
krt
|tr[td}|dur[ztt|dd d!|j|jd"}W n ttfyZ   Y nw t |r}t!|^}}t" fd#d$|D d}|dur}|||< q| ||< q|S )%a  Iterate over the namespace and:
    * gather private attributes
    * check for items which look like fields but are not (e.g. have no annotation) and warn.

    Args:
        namespace: The attribute dictionary of the class to be created.
        ignored_types: A tuple of ignore types.
        base_class_vars: A set of base class class variables.
        base_class_fields: A set of base class fields.

    Returns:
        A dict contains private attributes info.

    Raises:
        TypeError: If there is a `__root__` field in model.
        NameError: If private attribute name is invalid.
        PydanticUserError:
            - If a field does not have a type annotation.
            - If a field on base class was overridden by a non-annotated attribute.
    r   )r4   r6   r   __root__zUTo define root models, use `pydantic.RootModel` rather than a field called '__root__'ri   r   rO   rP   	functools__zXPrivate attributes must not use dunder names; use a single underscore prefix instead of r   zJPrivate attributes must not use valid field names; use sunder names, e.g. _z instead of my_fieldz>Fields must not use names with leading underscores; e.g., use )r   zField z defined on a base class was overridden by a non-annotated attribute. All field definitions, including overrides, require a type annotation.zmodel-field-overridden)codez requires a type annotationzmodel-field-missing-annotationz)A non-annotated attribute was detected: `z = z3`. All model fields require a type annotation; if `z` is not meant to be a field, you may be able to resolve this error by annotating it as a `ClassVar` or updating `model_config['ignored_types']`.NFT)is_argumentis_class)globalnslocalnsc                 3  s    | ]
}t | r|V  qd S rw   )rA   )ry   r;   r4   rL   rM   r{     s    z$inspect_namespace.<locals>.<genexpr>)#fieldsr4   r6   r&   default_ignored_typesr@   r   r   r   r   rA   rh   rO   rP   
startswithrK   add	NameErrorr    lstripr!   r.   r   r   r   r:   sys	_getframer,   r+   	f_globalsf_localsr-   r   next)r_   r   r   r   r6   r3   all_ignored_typesr   raw_annotationsignored_namesvar_namevaluesuggested_nameann_nameann_typeframer   metadatar   rL   r  rM   r   e  s   
















r   r   rq   c                 C  s@   t |d}t| }|d tjhv st|dd |jkr|| _d S d S )Nr   __code__)r   make_hash_funcr<   r   r   r  )r   r]   base_hash_funcnew_hash_funcrL   rL   rM   r     s
   
 
r   c                   s0   | j rtj| j   ndd  d	 fdd}|S )
Nc                 S  rV   )Nr   rL   )r   rL   rL   rM   <lambda>  rX   z make_hash_func.<locals>.<lambda>rH   r   r=   intc                   s6   zt  | jW S  ty   t  t| j Y S w rw   )hashr   KeyErrorr1   r   getterrL   rM   	hash_func  s
   z!make_hash_func.<locals>.hash_func)rH   r   r=   r!  )r   operator
itemgetterr   )r   r&  rL   r$  rM   r    s   
r  r   r   r   NsResolver | Nonec           	      C  sl   t | }t| ||||d\}}|| _| j| |D ]}| j|d}|dur3|jtur3t	| ||j qdS )a<  Collect and set `cls.__pydantic_fields__` and `cls.__class_vars__`.

    Args:
        cls: BaseModel or dataclass.
        bases: Parents of the class, generally `cls.__bases__`.
        config_wrapper: The config wrapper instance.
        ns_resolver: Namespace resolver to use when getting model annotations.
    )typevars_mapN)
r$   r   r   ro   r   rp   r   r   r   setattr)	r   r]   r   r   r*  r  r   r9   r  rL   rL   rM   r   
  s   r   r   r\   r:   r   rd   r   rf   c             
   C  s8  |j r
t| | dS t| }t|||}tt|jdd|dd}z| | |}	W n! tyI }
 z|r3 t| |d|
j	 d W Y d}
~
dS d}
~
ww |j
| jd}z||	}	W n |jyh   t| | Y dS w |	| _t|	| |ps| j| j|rydnd	||j| _t|	|| _d
| _tdtt| j| j|j|jd| _d
S )a&  Finish building a model class.

    This logic must be called after class has been created since validation functions must be bound
    and `get_type_hints` requires a class object.

    Args:
        cls: BaseModel or dataclass.
        cls_name: The model or dataclass name.
        config_wrapper: The config wrapper instance.
        raise_errors: Whether to raise errors.
        ns_resolver: The namespace resolver instance to use during schema building.
        create_model_module: The module of the class to be created, if created by `create_model`.

    Returns:
        `True` if the model is successfully completed, else `False`.

    Raises:
        PydanticUndefinedAnnotation: If `PydanticUndefinedAnnotation` occurs in`__get_pydantic_core_schema__`
            and `raise_errors=True`.
    F)from_dunder_get_core_schemaunpack)ref_moder?   N)titlecreate_modelr7   T__signature__)rT   r  populate_by_nameextra)defer_buildr'   r$   r"   r)   r   generate_schema__get_pydantic_core_schema__r   r   core_configrN   clean_schemaCollectedInvalid__pydantic_core_schema__r   rO   rP   plugin_settingsr   r   __pydantic_serializer__r   r0   r*   r   r   r2  r3  r1  )r   r\   r   r   r   r   r*  
gen_schemahandlerschemaer7  rL   rL   rM   r   +  sj   


	
r   c                 C  s   | j  D ]\}}|j }dur t|}|| | t| || q| j D ]%\}}|j }durKtt|j	dsKt||j	}|| | t| || q&dS )z8Set data descriptors on the class for deprecated fields.N__deprecated__)
r   r   deprecation_message_DeprecatedFieldDescriptorr   r+  r   r   r   wrapped_property)r   field
field_infomsgdesccomputed_field_inforL   rL   rM   r     s   r   c                   @  sF   e Zd ZU dZded< ddd
dZdddZddddZd ddZdS )!rC  aW  Read-only data descriptor used to emit a runtime deprecation warning before accessing a deprecated field.

    Attributes:
        msg: The deprecation message to be emitted.
        wrapped_property: The property instance if the deprecated field is a computed field, or `None`.
        field_name: The name of the field being deprecated.
    r:   
field_nameNrG  rD  property | Noner=   r>   c                 C  s   || _ || _d S rw   )rG  rD  )rH   rG  rD  rL   rL   rM   r     s   
z#_DeprecatedFieldDescriptor.__init__r   rq   r   c                 C  s
   || _ d S rw   )rJ  )rH   r   r   rL   rL   rM   r        
z'_DeprecatedFieldDescriptor.__set_name__r   BaseModel | Noneobj_typetype[BaseModel] | Noner   c                 C  sb   |d u r| j d ur| j d |S t| jtj| jtjdd | j d ur+| j ||S |j	| j S )Nr   rr   )
rD  __get__r   rJ  rB   rC   rG  builtinsDeprecationWarningr   )rH   r   rN  rL   rL   rM   rP    s   


z"_DeprecatedFieldDescriptor.__get__r  r   c                 C  s
   t | jrw   )r   rJ  )rH   r   r  rL   rL   rM   __set__  rL  z"_DeprecatedFieldDescriptor.__set__rw   )rG  r:   rD  rK  r=   r>   )r   rq   r   r:   r=   r>   )r   rM  rN  rO  r=   r   )r   r   r  r   r=   r   )	rN   rO   rP   rQ   r   r   r   rP  rS  rL   rL   rL   rM   rC    s   
 
rC  c                   @  s.   e Zd ZdZdddZdddZdd
dZdS )_PydanticWeakRefa  Wrapper for `weakref.ref` that enables `pickle` serialization.

    Cloudpickle fails to serialize `weakref.ref` objects due to an arcane error related
    to abstract base classes (`abc.ABC`). This class works around the issue by wrapping
    `weakref.ref` instead of subclassing it.

    See https://github.com/pydantic/pydantic/issues/6763 for context.

    Semantics:
        - If not pickled, behaves the same as a `weakref.ref`.
        - If pickled along with the referenced object, the same `weakref.ref` behavior
          will be maintained between them after unpickling.
        - If pickled without the referenced object, after unpickling the underlying
          reference will be cleared (`__call__` will always return `None`).
    r   r   c                 C  s"   |d u r	d | _ d S t|| _ d S rw   )_wrweakrefref)rH   r   rL   rL   rM   r     s   
z_PydanticWeakRef.__init__r=   c                 C  s   | j d u rd S |   S rw   )rU  r   rL   rL   rM   __call__  s   
z_PydanticWeakRef.__call__4tuple[Callable, tuple[weakref.ReferenceType | None]]c                 C  s   t |  ffS rw   )rT  r   rL   rL   rM   
__reduce__  s   z_PydanticWeakRef.__reduce__N)r   r   )r=   r   )r=   rY  )rN   rO   rP   rQ   r   rX  rZ  rL   rL   rL   rM   rT    s
    

rT  ddict[str, Any] | Nonec              	   C  sR   | du rdS i }|   D ]\}}zt|}W n ty!   |}Y nw |||< q|S )aX  Takes an input dictionary, and produces a new value that (invertibly) replaces the values with weakrefs.

    We can't just use a WeakValueDictionary because many types (including int, str, etc.) can't be stored as values
    in a WeakValueDictionary.

    The `unpack_lenient_weakvaluedict` function can be used to reverse this operation.
    N)r   rT  r   )r[  resultr9   r;   proxyrL   rL   rM   r     s   
r   c                 C  sP   | du rdS i }|   D ]\}}t|tr!| }|dur |||< q|||< q|S )zAInverts the transform performed by `build_lenient_weakvaluedict`.N)r   rA   rT  )r[  r]  r9   r;   rL   rL   rM   r     s   

r   )maxsizec                  C  s<   ddl m}  ttttt| tg}tj	dkr|
tj t|S )Nr   )r2   )      )r  r2   r   r   r   r   r   r   r  version_infoappendr   r   )r2   r   rL   rL   rM   r    s   

r  )rT   rU   r=   r   r   )r_   r`   r]   r^   r=   r   )
r_   r`   r   r^   r   r   r   r   r=   r   )r   rq   r]   r^   r=   r>   )r   rq   r=   r   )
r   rq   r]   r^   r   r   r   r)  r=   r>   )r   rq   r\   r:   r   r   r   rd   r   r)  r   rf   r=   rd   )r   rq   r=   r>   )r[  r\  r=   r\  )r=   r^   )jrQ   
__future__r   _annotationsrQ  r'  r  r   rB   rV  abcr   r   r   r   typesr   r   r   r	   r
   r   r   pydantic_corer   r   typing_extensionsr   r   r   r   errorsr   r   plugin._schema_validatorr   r   r   _configr   _decoratorsr   r   r   r   _fieldsr   r    r!   _generate_schemar"   	_genericsr#   r$   _import_utilsr%   r&   _mock_val_serr'   _namespace_utilsr(   _schema_generation_sharedr)   
_signaturer*   _typing_extrar+   r,   r-   r.   r/   _utilsr0   r1   r   r  r2   r3   r4   r5   PydanticModelFieldr6   PydanticModelPrivateAttrmainr7   rR  r<   __setattr__r   r   r8   rW   r[   rk   r   r   r   r  r   r   r   rC  rT  r   r   r  rL   rL   rL   rM   <module>   s|     
 
x

 
	

&
]%
#
