U
    nËj»K  ã                   @   sÆ  U d dl Z d dlZd dlmZ d dlmZ d dlmZmZmZmZ	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZm Z!m"Z# zd dlm$Z% W n  e&k
rÖ   d dlm'Z% Y nX zd dlm(Z) W n e&k
�r   d	Z)Y nX zd d
l*m+Z, W n e&k
�r.   d	Z,Y nX e j-dk �rReeeedœdd„Z.neeeedœdd„Z.e j-dk �rxeZ/ndjeeeedœdd„Z/edƒZ0e	def Z1e	g ef Z2e	de0f Z3ddhZ4ehZ5ee e6d< e7edƒ�rìe5 8ej¡ e j-dk �ree eee  dœdd„Z9n(d dlm9Z: ee eee  dœdd„Z9e j-dk �rtd d lm;Z; ee eedf dœd!d"„Z<nDd d#lm<Z= ee eedf dœd$d%„Z>ee eedf dœd&d"„Z<e j-dk �rÞee ee dœd'd(„Z?n0d d)lm@Z@ d d*lmAZA ee ee dœd+d(„Z?e j-d,k �r:eee  eBdœd-d.„ZCe)fZDn8d dl*Z*d dlZeee  eBdœd/d.„ZCej;e*j(e*j+fZDeeEef ZFe�r>d d0lGmHZH eeeEef ddf ZIeeEef ZJeeef ZKeeE ZLeeE ZMeeNeEf ZOeeO ZPeeOef ZQeeOef ZRee1ddf ZSeeeeE ef  ZTd1ZUeU�r0eVe ZWneVeeef ZWd2ZXdjYZZdeZed fZ[eeeef e6d3< e j-dk �r†eeBd4œd5d6„Z\n6e j-dd7… dk�r¬eeBd4œd8d6„Z\neeBd4œd9d6„Z\ee eEd:œd;d<„Z]eeEee f eeE eeEee f d=œd>d?„Z^ee eBd4œd@dA„Z_ee eBd4œdBdC„Z`ee eedf d4œdDdE„Zaee eedf d4œdFdG„Zbee eBd4œdHdI„Zcee eBd4œdJdK„ZdeeBd4œdLdM„ZeeeBd4œdNdO„ZfedPeEƒZgee eBd4œdQdR„Zhee ee d4œdSdT„Zieee  eBd:œdUdV„Zjeee  eBd:œdWdX„Zkee eBdYœdZd[„Zlee eBdYœd\d]„Zmd^eedd_œd`da„Zndkee ed^ eeee eEef e1f dbeeeo df ddcœddde„Zpee edeBee f d4œdfdg„Zqeee dœdhdi„ZrdS )lé    N)ÚCallable)ÚPathLike)ÚTYPE_CHECKINGÚAbstractSetÚAnyr   ÚClassVarÚDictÚ
ForwardRefÚ	GeneratorÚIterableÚListÚMappingÚNewTypeÚOptionalÚSequenceÚSetÚTupleÚTypeÚTypeVarÚUnionÚ
_eval_typeÚcastÚget_type_hints)Ú	AnnotatedÚFinalÚLiteralÚNotRequiredÚRequired)Ú_TypingBase)Ú_Final)ÚGenericAlias© )Ú	UnionType©é   é	   )Útype_ÚglobalnsÚlocalnsÚreturnc                 C   s   |   ||¡S ©N)Ú	_evaluate©r&   r'   r(   r!   r!   ú6/tmp/pip-unpacked-wheel-zvl9l234/pydantic/v1/typing.pyÚevaluate_forwardref:   s    r.   c                 C   s   t t| ƒj||tƒ d�S )N)Zrecursive_guard)r   r   r+   Úsetr,   r!   r!   r-   r.   ?   s    )Úobjr'   r(   r)   c                 C   s   t | ||dd�S )NT)Zinclude_extras)r   )r0   r'   r(   r!   r!   r-   Úget_all_type_hintsO   s    r1   Ú_T.ZAnnotatedMetaÚ_AnnotatedAliasÚLITERAL_TYPESr   ©r$   é   )Útr)   c                 C   s(   t | ƒjtkrttt tƒS t| dd ƒS )NÚ
__origin__)ÚtypeÚ__name__ÚAnnotatedTypeNamesr   r   r   r   Úgetattr)r7   r!   r!   r-   Ú
get_originh   s    r=   )r=   )Útpr)   c                 C   s0   t | ƒjtkrttt tƒS t| ƒp.t| ddƒS )a4  
        We can't directly use `typing.get_origin` since we need a fallback to support
        custom generic classes like `ConstrainedList`
        It should be useless once https://github.com/cython/cython/issues/3537 is
        solved and https://github.com/pydantic/pydantic/pull/1753 is merged.
        r8   N)	r9   r:   r;   r   r   r   r   Ú_typing_get_originr<   ©r>   r!   r!   r-   r=   q   s    )Ú_GenericAliasc                 C   sl   t | ƒjtkr| j| j S t| tƒr`| j}| jtkr\|r\|d t	k	r\t
|dd… ƒ|d f}|S t| ddƒS )z²Compatibility version of get_args for python 3.7.

        Mostly compatible with the python 3.8 `typing` module version
        and able to handle almost all use cases.
        r   NéÿÿÿÿÚ__args__r!   )r9   r:   r;   rC   Ú__metadata__Ú
isinstancerA   r8   r   ÚEllipsisÚlistr<   )r7   Úresr!   r!   r-   Úget_args€   s    
rI   ©rI   c                 C   s\   t | dƒrtf| j S z,| td ks:tjdkr@| td kr@W dS W n tk
rV   Y nX dS )aI  
        In python 3.9, `typing.Dict`, `typing.List`, ...
        do have an empty `__args__` by default (instead of the generic ~T for example).
        In order to still support `Dict` for example and consider it as `Dict[Any, Any]`,
        we retrieve the `_nparams` value that tells us how many parameters it needs.
        Ú_nparamsr!   r#   )r!   )Úhasattrr   rK   r   ÚsysÚversion_infoÚtupleÚ	TypeErrorr@   r!   r!   r-   Ú_generic_get_args’   s    
"
rQ   c                 C   s6   t | ƒjtkr| j| j S t| ƒp4t| ddƒp4t| ƒS )aÆ  Get type arguments with all substitutions performed.

        For unions, basic simplifications used by Union constructor are performed.
        Examples::
            get_args(Dict[str, int]) == (str, int)
            get_args(int) == ()
            get_args(Union[int, Union[T, int], str][int]) == (int, str)
            get_args(Union[int, Tuple[T, int]][str]) == (int, Tuple[str, int])
            get_args(Callable[[], T][int]) == ([], int)
        rC   r!   )r9   r:   r;   rC   rD   Ú_typing_get_argsr<   rQ   r@   r!   r!   r-   rI   ¦   s    c                 C   s   | S )zÛPython 3.9 and older only supports generics from `typing` module.
        They convert strings to ForwardRef automatically.

        Examples::
            typing.List['Hero'] == typing.List[ForwardRef('Hero')]
        r!   r@   r!   r!   r-   Úconvert_generics¹   s    rS   )Ú_UnionGenericAlias)r3   c                    s¼   t ˆ ƒ}|rtˆ dƒsˆ S tˆ ƒ}|tkrDtt|d ƒ|dd… ƒS t‡ fdd„|D ƒƒ}||krfˆ S tˆ tƒrzt||ƒS tˆ t	ƒrŽt
||ƒS ztˆ d|ƒ W n tk
r²   Y nX ˆ S dS )aì  
        Recursively searches for `str` type hints and replaces them with ForwardRef.

        Examples::
            convert_generics(list['Hero']) == list[ForwardRef('Hero')]
            convert_generics(dict['Hero', 'Team']) == dict[ForwardRef('Hero'), ForwardRef('Team')]
            convert_generics(typing.Dict['Hero', 'Team']) == typing.Dict[ForwardRef('Hero'), ForwardRef('Team')]
            convert_generics(list[str | 'Hero'] | int) == list[str | ForwardRef('Hero')] | int
        rC   r   é   Nc                 3   s2   | ]*}t |tƒr"t ˆ tƒr"t|ƒnt|ƒV  qd S r*   )rE   ÚstrÚTypingGenericAliasr	   rS   )Ú.0Úargr@   r!   r-   Ú	<genexpr>Ü   s   ÿz#convert_generics.<locals>.<genexpr>)r=   rL   rI   r   r3   rS   rO   rE   rW   ÚTypesUnionTyperT   ÚsetattrÚAttributeError)r>   ÚoriginÚargsZ	convertedr!   r@   r-   rS   Ç   s(    
þ



©r$   é
   c                 C   s   | t kS r*   )r   r@   r!   r!   r-   Úis_unionò   s    rb   c                 C   s   | t kp| tjkS r*   )r   Útypesr"   r@   r!   r!   r-   rb   û   s    )Ú
ModelFieldF)'ÚAnyCallableÚNoArgAnyCallableÚNoneTypeÚis_none_typeÚdisplay_as_typeÚresolve_annotationsÚis_callable_typeÚis_literal_typeÚall_literal_valuesÚis_namedtupleÚis_typeddictÚis_typeddict_specialÚis_new_typeÚnew_type_supertypeÚis_classvarÚis_finalvarÚupdate_field_forward_refsÚupdate_model_forward_refsÚTupleGeneratorÚ
DictStrAnyÚDictAnyÚSetStrÚListStrÚIntStrÚAbstractSetIntStrÚDictIntStrAnyÚCallableGeneratorÚReprArgsÚAnyClassMethodr   ÚWithArgsTypesrI   r=   Úget_sub_typesÚtyping_baser1   rb   ÚStrPathÚMappingIntStrAnyÚ
NONE_TYPES)r&   r)   c                 C   s   | t kS r*   ©r‡   ©r&   r!   r!   r-   rh   R  s    rh   é   c                 C   s0   t D ]}| |kr dS qt| ƒr,t| ƒdkS dS )NTr*   F)r‡   rl   rm   )r&   Z	none_typer!   r!   r-   rh   W  s    c                 C   s   | t kS r*   rˆ   r‰   r!   r!   r-   rh   d  s    )Úvr)   c                 C   s˜   t | tƒs$t | tƒs$t | tƒs$| j} tt| ƒƒrLdd tt	t
| ƒƒ¡› d�S t | tƒrft| ƒ dd¡S z| jW S  tk
r’   t| ƒ dd¡ Y S X d S )NzUnion[z, ú]ztyping.Ú )rE   r„   r‚   r9   Ú	__class__rb   r=   ÚjoinÚmapri   rI   rV   Úreplacer:   r]   ©r‹   r!   r!   r-   ri   h  s    
ri   )Úraw_annotationsÚmodule_namer)   c              	   C   sÞ   d}|r2zt j| }W n tk
r*   Y nX |j}i }|  ¡ D ]š\}}t|tƒrŽdt j  krfdksrn t jdkr‚t|ddd�}nt|dd�}z,t jd	kr¬t	||dd
d�}nt	||dƒ}W n t
k
rÎ   Y nX |||< q>|S )z‚
    Partially taken from typing.get_type_hints.

    Resolve string or ForwardRef annotations into type objects if possible.
    Nr`   )r$   r%   r6   )r$   ra   rU   FT)Úis_argumentZis_class)r•   )r$   é   r!   )Ztype_params)rM   ÚmodulesÚKeyErrorÚ__dict__ÚitemsrE   rV   rN   r	   r   Ú	NameError)r“   r”   Úbase_globalsÚmoduleÚannotationsÚnameÚvaluer!   r!   r-   rj   z  s*    
"

rj   c                 C   s   | t kpt| ƒt kS r*   )r   r=   r‰   r!   r!   r-   rk   �  s    rk   c                 C   s   t d k	ot| ƒtkS r*   )r   r=   r4   r‰   r!   r!   r-   rl   ¡  s    rl   c                 C   s   t | ƒS r*   rJ   r‰   r!   r!   r-   Úliteral_values¥  s    r¡   c                 C   s(   t | ƒs| fS t| ƒ}tdd„ |D ƒƒS )zÐ
    This method is used to retrieve all Literal values as
    Literal can be used recursively (see https://www.python.org/dev/peps/pep-0586)
    e.g. `Literal[Literal[Literal[1, 2, 3], "foo"], 5, None]`
    c                 s   s    | ]}t |ƒD ]
}|V  qqd S r*   )rm   )rX   r    Úxr!   r!   r-   rZ   ³  s     
  z%all_literal_values.<locals>.<genexpr>)rl   r¡   rO   )r&   Úvaluesr!   r!   r-   rm   ©  s    rm   c                 C   s    ddl m} || tƒot| dƒS )zy
    Check if a given class is a named tuple.
    It can be either a `typing.NamedTuple` or `collections.namedtuple`
    r   ©Úlenient_issubclassÚ_fields)Úpydantic.v1.utilsr¥   rO   rL   ©r&   r¥   r!   r!   r-   rn   ¶  s    rn   c                 C   s    ddl m} || tƒot| dƒS )zÉ
    Check if a given class is a typed dict (from `typing` or `typing_extensions`)
    In 3.10, there will be a public method (https://docs.python.org/3.10/library/typing.html#typing.is_typeddict)
    r   r¤   Ú	__total__)r§   r¥   ÚdictrL   r¨   r!   r!   r-   ro   À  s    ro   c                 C   s   | t kp| tkS r*   )ÚTypedDictRequiredÚTypedDictNotRequiredr‰   r!   r!   r-   Ú_check_typeddict_specialÊ  s    r­   c                 C   s   t | ƒpt t| ƒƒS )zN
    Check if type is a TypedDict special form (Required or NotRequired).
    )r­   r=   r‰   r!   r!   r-   rp   Î  s    rp   Ú	test_typec                 C   s   t | tjƒot| dƒS )z>
    Check whether type_ was created using typing.NewType
    Ú__supertype__)rE   r®   rŽ   rL   r‰   r!   r!   r-   rq   Ø  s    rq   c                 C   s   t | dƒr| j} q | S )Nr¯   )rL   r¯   r‰   r!   r!   r-   rr   ß  s    
rr   c                 C   s(   | d krdS | j tj ko&t| dd ƒdkS )NFÚ_namer   )rŽ   r   r<   r’   r!   r!   r-   Ú_check_classvarå  s    r±   c                 C   s2   | dkrdS | j tj ko0tjdk p0t| ddƒdkS )z9
    Check if a given type is a `typing.Final` type.
    NFr5   r°   r   )rŽ   r   rM   rN   r<   r’   r!   r!   r-   Ú_check_finalvarì  s    r²   )Úann_typer)   c                 C   s6   t | ƒst t| ƒƒrdS | jtkr2| j d¡r2dS dS )NTz	ClassVar[F)r±   r=   rŽ   r	   Ú__forward_arg__Ú
startswith©r³   r!   r!   r-   rs   ö  s
    rs   c                 C   s   t | ƒpt t| ƒƒS r*   )r²   r=   r¶   r!   r!   r-   rt     s    rt   rd   )Úfieldr'   r(   r)   c                 C   sŽ   d}| j jtkr(d}t| j ||p"dƒ| _ | jjtkrLd}t| j||pFdƒ| _|rX|  ¡  | jrx| jD ]}t|||d� qd| jdk	rŠ|  	¡  dS )z]
    Try to update ForwardRefs on fields based on this ModelField, globalns and localns.
    FTN©r'   r(   )
r&   rŽ   r	   r.   Zouter_type_ÚprepareZ
sub_fieldsru   Zdiscriminator_keyZ&prepare_discriminated_union_sub_fields)r·   r'   r(   r¹   Zsub_fr!   r!   r-   ru     s    

ru   rx   )ÚmodelÚfieldsÚjson_encodersr(   Úexc_to_suppressr)   c           
   	   C   sÜ   | j tjkr tj| j  j ¡ }ni }| | j| ¡ |D ],}zt|||d� W q6 |k
r`   Y q6X q6t| 	¡ ƒD ]f}t
|tƒrˆt|ƒ}nt
|tƒrp|}nqpzt|||p¨dƒ}	W n |k
rÆ   Y qpY nX | |¡||	< qpdS )zL
    Try to update model fields ForwardRefs based on model and localns.
    r¸   N)Ú
__module__rM   r—   r™   ÚcopyÚ
setdefaultr:   ru   r/   ÚkeysrE   rV   r	   r.   Úpop)
rº   r»   r¼   r(   r½   r'   ÚfÚkeyÚfrZnew_keyr!   r!   r-   rv     s(    




rv   c                 C   sF   | t krdS t| ƒdkrdS t| ƒ}|r6t|d t ƒs:dS |d S dS )z„
    Tries to get the class of a Type[T] annotation. Returns True if Type is used
    without brackets. Otherwise returns None.
    TNr   )r9   r=   rI   rE   )r&   r_   r!   r!   r-   Ú	get_classC  s    rÆ   c                 C   sD   t | ƒ}|tkr tt| ƒd ƒS t|ƒr:dd„ t| ƒD ƒS | gS dS )z~
    Return all the types that are allowed by type `tp`
    `tp` can be a `Union` of allowed types or an `Annotated` type
    r   c                 S   s   g | ]}t |ƒD ]}|‘qqS r!   )rƒ   )rX   r7   r¢   r!   r!   r-   Ú
<listcomp>^  s     
  z!get_sub_types.<locals>.<listcomp>N)r=   r   rƒ   rI   rb   )r>   r^   r!   r!   r-   rƒ   U  s    rƒ   )NN)r!   )srM   ÚtypingÚcollections.abcr   Úosr   r   r   r   ZTypingCallabler   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Ztyping_extensionsr   r   r   r   r¬   r   r«   r   r„   ÚImportErrorr   r    rW   rc   r"   r[   rN   r.   r1   r2   re   rf   ZAnyArgTCallabler;   r4   Ú__annotations__rL   Úaddr=   r?   rA   rI   rR   rQ   rS   rT   r3   Úboolrb   r‚   rV   r…   Zpydantic.v1.fieldsrd   rw   rx   ry   rz   r{   Úintr|   r}   r~   r†   r   r€   ZMYPYÚclassmethodr�   Ú__all__rŽ   rg   r‡   rh   ri   rj   rk   rl   r¡   rm   rn   ro   r­   rp   r®   rq   rr   r±   r²   rs   rt   ru   ÚBaseExceptionrv   rÆ   rƒ   r!   r!   r!   r-   Ú<module>   sÚ    `


)
+ .#



 ûú'"