iAddlmZddlZddlmZddlZddlZddlZddlm 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 ` ````` rArgrg|sE.%*0/ *c""  H' 4FGGJ$ *Gc   C z " "C:&&C  EEM  & & ' &   j ) ) )          #+      -ZBBB r@GenericProtocol[_T]rvr7rDcft|dr||S|j|S)N copy_with)hasattrryrE)rYrvs rArtrts:z;''/##H---$X..r@rc expression module_nameOptional[Mapping[str, Any]]rdOptional[Type[Any]]rc tj|j}n(#t$r}t d|d||d}~wwxYw |Vt |j}||j|||t|||}nt|||}|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 NameErrordict setdefaultr:updateeval Exception) r{r|r]rd base_globalske cls_namespacerYerrs rArnrns#'*{;'?'H  0k 0 0#- 0 0       !233M  $ $X%6 A A A   . . .j-AAJJj,@@J   ? ? ?   s) ?:?A)B.. C8C  Cr]namecbd|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.rrz2 isn't present in sys.modules; can't resolve name zCould not locate name z in module )rnrrrrrbuiltins)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_namersP.T;//sJ---  s # 22rTc@tfd|jDS)Nc 8g|]}t|iS)rrf)rhannorWr[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_pep593r2s   F!25!9!9Y!FFr@rTypeGuard[Iterable[Any]]cpt|tjot|ttf SrK)rmcollections_abcr r\bytes)rs rAis_non_string_iterabler6s6 c?3 4 4 Z c5\>>:r@c.t|tuSrK)rr rs rArprp<s e   ''r@TypeGuard[NewType]c"t|dS)N __supertype__rzrs rA is_newtyper@s 5/ * **r@TypeGuard[GenericProtocol[Any]]cBt|dot|dS)Nr8rErrs rAroroHs! 5* % % F'%*F*FFr@TypeGuard[TypeAliasType]c,t|tSrK)rmr)rs rA is_pep695rLs e] + ++r@r c^|j}t|r|j}t||SrK)rr)r super_types rAflatten_newtyperPs:$J Z .-  Z . r@ check_genericTypeGuard[ForwardRef]ct|trdS|r-t|rtd|jDSdS)NTc36K|]}t|dVdS)TN)rk)rhargs rArjzis_fwd_ref..]s,CCS:c4((CCCCCCr@F)rmr roanyr8)rrs rArkrkWsV%$$t :e,,CCENCCCCCCur@cdSrKr?rs rAde_optionalize_union_typesrbs363r@cdSrKr?rs rArrfs?Bsr@cdSrKr?rs rArrjs #r@ct|rt|St|rQt|j}|t |tt|S|S)zmGiven a type, filter out ``Union`` types that include ``NoneType`` to not include the ``NoneType``. ) rk"de_optionalize_fwd_ref_union_types is_optionalrqr8discardr3NoneFwdr)rtyps rArrpsw% 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)rhris rArz6de_optionalize_fwd_ref_union_types..s%JJJt46>>*T"">>>r@z\s*\|\s*r4|c3&K|] }|dk|V dS)r4Nr?)rhps rArjz5de_optionalize_fwd_ref_union_types..s&"I"IQ&[[1[[[["I"Ir@)rlrematchgroupr splitrjoin)rrYmmrv 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_unionrqr8rr3rs)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)rrmtype 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@rArrs4????====444444r@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__rNr@rr4cdSrKr?rs rArzDescriptorReference.__set__rNr@cdSrKr?rs rArzDescriptorReference.__delete__rNr@N)rrrrrHr)rrrrrHr4rr:r;r<r=rrrrr?r@rAr r sR9EEEECCCC888888 99r@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 rrrrrHrcdSrKr?rs rArzRODescriptorReference.__get__(rNr@rrcdSrKr?rs rArzRODescriptorReference.__set__*rNr@cdSrKr?rs rArz RODescriptorReference.__delete__,rNr@N)rrrrrHr)rrrrrHr)rrrHrrr?r@rArrsR=HHHHEEEE<<<<<< ==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__=rNr@rr4cdSrKr?rs rArzCallableReference.__set__?rNr@cdSrKr?rs rArzCallableReference.__delete__ArNr@N)rrrrrHr)rrrrrHr4rrr?r@rArr2sR9CCCCAAAA888888 99r@r)rWrXrYrZr[r\r]r^rTr_rUr`rVrarHrX)rYrwrvr7rHrD) r{r\r|r\r]r}rdr~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 _LiteralStarrgrtrnrrrrrrprrorrrkrrrrrrrrrrrrr rrrrr?r@rAr"s#"""""))))))  A888888<<<<<<100000000000666666::::::444444<<<<<<888888666666@@@@@@888888888888888888......@@@@@@ WT genn T * * * W\ 6 6 6  genn T * * * <+******tDzzH *V  IsJ8NN .....x...hrl(33333Xc6k2333s| ;?!(,CCCCCCL////,0$( !!!!!!P,0 D....;? ,GGGG ((((++++GGGG,,,,7< 666 6 BBB B   .>////IN(EEEE(((( F F F F6:      55555h555 ///99999'%.999( 7:_ E E E=====GH-==="ge8HS#X$6788899999 99999r@