e?ddlmZddlZddlmZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZ ddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%dd lm&Z&dd!lm'Z'dd"lm(Z(dd#lm)Z)dd$lm*Z*dd%lm+Z+dd&lm,Z,dd'lm-Z-dd(lm.Z.ed)e *Z/ed+Z0ed,d-Z1ed.d/Z2ed0Z3ed1d-Z4ej5rdd2l6m7Z7n e8dZ7e d3Z9e#Z:e$Z;eee eGd7d8e(e/Z?Gd9d:e(e0e4fZ@e%d;ZAddz*de_stringify_annotation..s]    $"- /+          r@) is_fwd_ref__forward_arg__ isinstancer\eval_expression is_generic is_literalsetaddtupler8_copy_generic_annotation_with) rWrYr[r]rTrUrVoriginal_annotationelementss ` ````` rArfrf~sC.%*0/ *c""  H' 4FGGJ$ *G   C z " "C:&&C  EEM  & & ' &   j ) ) )          #+      -ZBBB r@GenericProtocol[_T]rur7rDcft|dr||S|j|S)N copy_with)hasattrrxrE)rYrus rArsrss:z;''/##H---$X..r@rc expression module_nameOptional[Mapping[str, Any]]rc tj|j}n(#t$r}t d|d||d}~wwxYw t |||}|S#t $r}t d||d}~wwxYw)NModule z9 isn't present in sys.modules; can't evaluate expression z"Could not de-stringify annotation )sysmodules__dict__KeyError NameErroreval Exception)rzr{r] base_globalskerYerrs rArmrms '*{;'?'H  0k 0 0#- 0 0    *lG<<   ? ? ?   s( ?:?A A8 A33A8namecbd|vrt|||S tj|j}n(#t$r}t d|d||d}~wwxYw ||S#t$rC} t j|cYd}~S#t$rYnwxYwt d|d||d}~wwxYw)N.rcr~z2 isn't present in sys.modules; can't resolve name zCould not locate name z in module )rmrrrrrbuiltins)rr{r]rrs rAeval_name_onlyrs'  d{{t['BBBB'*{;'?'H  #k # #  # #    D!!     $T* * * * * * *    D  CT C Ck C C    sJ0 AAAA!! B.,B=B. B B)BB))B.cj t||}t|d|S#t$r|cYSwxYw)Nr:)rgetattrr)rr{objs rAresolve_name_to_real_class_namer sP.T;//sJ---  s # 22rTc@tfd|jDS)Nc 8g|]}t|iS)rre)rgannorWr[rTs rA z/de_stringify_union_elements..sI  $"-     r@)make_union_typer8)rWrYr[r]rTs` ` `rAde_stringify_union_elementsrsI  #+   r@type_Optional[_AnnotationScanType]c6|duot|tuSrK)typing_get_originrrs rA is_pep593r(s   F!25!9!9Y!FFr@rTypeGuard[Iterable[Any]]cpt|tjot|ttf SrK)rlcollections_abcr r\bytes)rs rAis_non_string_iterabler,s6 c?3 4 4 Z c5\>>:r@c.t|tuSrK)rr rs rAroro2s e   ''r@TypeGuard[NewType]c"t|dS)N __supertype__ryrs rA is_newtyper6s 5/ * **r@TypeGuard[GenericProtocol[Any]]cBt|dot|dS)Nr8rErrs rArnrn>s! 5* % % F'%*F*FFr@TypeGuard[TypeAliasType]c,t|tSrK)rlr)rs rA is_pep695rBs e] + ++r@r c^|j}t|r|j}t||SrK)rr)r super_types rAflatten_newtyperFs:$J Z .-  Z . r@ check_genericTypeGuard[ForwardRef]ct|trdS|r-t|rtd|jDSdS)NTc36K|]}t|dVdS)TN)rj)rgargs rArizis_fwd_ref..Ss,CCS:c4((CCCCCCr@F)rlr rnanyr8)rrs rArjrjMsV%$$t :e,,CCENCCCCCCur@cdSrKr?rs rAde_optionalize_union_typesrXCr@cdSrKr?rs rArr]rr@cdSrKr?rs rArrbs Cr@ct|rt|St|rQt|j}|t |tt|S|S)zmGiven a type, filter out ``Union`` types that include ``NoneType`` to not include the ``NoneType``. ) rj"de_optionalize_fwd_ref_union_types is_optionalrpr8discardr3NoneFwdr)rtyps rArrisw% 1%888 U   %.!! H G$$ r@r c|j}tjd|}|r|ddkr"t |dS|ddkr;tjd|d}t d|DS|Stjd|}d |vr,t d d |DS|S) zreturn the non-optional type for Optional[], Union[None, ...], x|None, etc. without de-stringifying forward refs. unfortunately this seems to require lots of hardcoded heuristics z^(.+?)\[(.+)\]$rrrz,\s*c8g|]}|dkt|S)r4r)rgrhs rArz6de_optionalize_fwd_ref_union_types..s%JJJt46>>*T"">>>r@z\s*\|\s*r4|c3&K|] }|dk|V dS)r4Nr?)rgps rAriz5de_optionalize_fwd_ref_union_types..s&"I"IQ&[[1[[[["I"Ir@)rkrematchgroupr splitrjoin)rrYmmru pipe_tokenss rArrs&J $j 1 1B  88A;;* $ $bhhqkk** * XXa[[G # #x!55H"JJxJJJ L(; 33K #(("I"Ik"I"I"IIIJJJ Lr@typesc\ttt|S)zMake a Union type. This is needed by :func:`.de_optionalize_union_types` which removes ``NoneType`` from a ``Union``. )rrrrQ)rs rArrs" U   ' ' . ..r@ include_union discard_noneTuple[Type[Any], ...]ct|rTt|j}|r|t|r|ft |zSt |S|fS)zPReturn a type as a tuple of individual types, expanding for ``Union`` types.)is_unionrpr8rr3rr)rrrrs rA expand_unionsrsi  %.!!  " KK ! ! !  8eCjj( (:: xr@TypeGuard[ArgsTypeProcotol]c&t|dddS)Nrr UnionType is_origin_ofrs rArrs     r@cLt|ott|vSrK)rr3typing_get_argsrs rAis_optional_unionrs" u   D(oe.D.D"DDr@c"t|dS)Nrrrs rArrs w ' ''r@ class_obj'Union[Tuple[Type[Any], ...], Type[Any]]crt|}|dSt|tot||S)z[return True if the given type has an __origin__ that shares a base with the given classNF)rrltype issubclass)rrorigins rAis_origin_of_clsrs: u % %F ~u fd # # E 69(E(EEr@)modulenamesr Optional[str]ct|}|dSt||vo|dup|j|S)z\return True if the given type has an __origin__ with the given name and optional module.NF)r_get_type_namer; startswith)rrrrs rArrsR u % %F ~u & ! !U * $>&+66v>>r@cttjr|jSt|dd}|t|dd}|S)Nr:_name)rpy310r:r)rtyp_names rArrsA |~5*d33  ugt44Hr@c&eZdZd dZd d Zdd Zd S)DescriptorProtoinstanceobjectownerrrHcdSrKr?rLrrs rA__get__zDescriptorProto.__get__rNr@valuer4cdSrKr?rLrrs rA__set__zDescriptorProto.__set__rNr@cdSrKr?rLrs rA __delete__zDescriptorProto.__delete__rNr@N)rrrrrHr)rrrrrHr4rrrHr4)r:r;r<rrrr?r@rArrsP              r@r_DESCc2eZdZdZerddZdd Zdd Zd Sd S)DescriptorReferenceaa descriptor that refers to a descriptor. used for cases where we need to have an instance variable referring to an object that is itself a descriptor, which typically confuses typing tools as they don't know when they should use ``__get__`` or not when referring to the descriptor assignment as an instance variable. See sqlalchemy.orm.interfaces.PropComparator.prop rrrrrHrcdSrKr?rs rArzDescriptorReference.__get__ Cr@rr4cdSrKr?rs rArzDescriptorReference.__set__rr@cdSrKr?rs rArzDescriptorReference.__delete__rr@N)rrrrrHr)rrrrrHr4rr:r;r<r=rrrrr?r@rArrsn                 r@r_DESC_co)r+r.c2eZdZdZerddZdd Zdd Zd Sd S)RODescriptorReferenceza descriptor that refers to a descriptor. same as :class:`.DescriptorReference` but is read-only, so that subclasses can define a subtype as the generically contained element rrrrrHr cdSrKr?rs rArzRODescriptorReference.__get__'rr@rrcdSrKr?rs rArzRODescriptorReference.__set__*rr@cdSrKr?rs rArz RODescriptorReference.__delete__-rr@N)rrrrrHr )rrrrrHr)rrrHrr r?r@rAr r sn                 r@r _FN.c2eZdZdZerddZdd Zdd Zd Sd S)CallableReferenceza descriptor that refers to a callable. works around mypy's limitation of not allowing callables assigned as instance variables rrrrrHrcdSrKr?rs rArzCallableReference.__get__?rr@rr4cdSrKr?rs rArzCallableReference.__set__Brr@cdSrKr?rs rArzCallableReference.__delete__Err@N)rrrrrHr)rrrrrHr4rr r?r@rArr4sn                 r@r)rWrXrYrZr[r\r]r^rTr_rUr`rVrarHrX)rYrvrur7rHrD)rzr\r{r\r]r|rHr)rr\r{r\r]r|rHr)rr\r{r\rHr\) rWrXrYr6r[r\r]r^rTr_rHrX)rrrHr`)rrrHr)rrZrHr`)rrrHr)rrZrHr)rrZrHr)rr rHrX)F)rrZrr`rHr)rr\rHr\)rrXrHrX)rrZrHrZ)rr rHrZ)rrZrHrX)FF)rrXrr`rr`rHr)rrrHr)rrrHr`)rrrrrHr`)rrrr\rrrHr`)rrXrHr\)a __future__rrcollections.abcabcrrrtypingrrrrr r r r r rrrrrrrrrrtyping_extensionsrrrrrrrr r!r"r#r$r%r&r'r(r)r*r,r-r/r1r2rrr3rrrrr\rZr6rCrG _LiteralStarrfrsrmrrrrrrorrnrrrjrrrrrrrrrrrrrr r rrr?r@rArs'#"""""))))))  A888888<<<<<<100000000000666666::::::444444<<<<<<888888666666@@@@@@888888888888888888......@@@@@@ WT genn T * * * W\ 6 6 6  genn T * * * <+******tDzzH *V  IsJ8NN .....x...hrl(     Xc6k2   s| ;?!(,CCCCCCL////,0 8,0 D....;? ,GGGG ((((++++GGGG,,,,7<       .>////IN(EEEE(((( F F F F6:           h    ///'%.. 7:_ E E EGH-(ge8HS#X$67888 r@