NgGddlmZddlZddlmZddlZddlZddlZddlm 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-Z4ej5re6ej%e%gZ7n e6e%gZ7ej8rdd2l9m:Z:n e;dZ:e d3Zeee e?e ee.d4fZ@Gd5d6e(ZAGd7d8e(e/ZBGd9d:e(e0e4fZCe%d;ZDddz*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 ` ````` rArgrgsE.%*0/ *c""  H' 4FGGJ$ *Gc   C z " "C:&&C  EEM  & & ' &   j ) ) )          #+      -ZBBB r@type_c t|rt|tttt jt jt jt j t j fvrhtj dt|sFt|tdt!|DS|S)zfCorrect dict['x', 'y'] into dict[ForwardRef('x'), ForwardRef('y')] and similar for list, set z3typing.(?:Dict|List|Set|.*Mapping|.*Sequence|.*Set)cZg|](}t|trt|n|)Sr?)rmr\r rhris rA z,fixup_container_fwd_refs..sD)34(=(=GJt$$$4r@)rotyping_get_origindictrqlistcollections_abc MutableSetMutableMappingMutableSequencer Sequencerematchrepr__class_getitem__rstyping_get_argsrws rAfixup_container_fwd_refsrs 5 e $ $     &  *  +  #  $   A4;;   (!''99  / 6 6      Lr@GenericProtocol[_T]rvr7rDcft|dr||S|j|S)N copy_with)hasattrrrE)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 NameErrorr} setdefaultr:updateeval Exception) rrr]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)rrr]rrs rAeval_name_onlyr"s'  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)rrobjs rAresolve_name_to_real_class_namerDsP.T;//sJ---  s # 22rTc@tfd|jDS)Nc 8g|]}t|iS)rrf)rhannorWr[rTs rAr{z/de_stringify_union_elements..VsI  $"-     r@)make_union_typer8)rWrYr[r]rTs` ` `rAde_stringify_union_elementsrMsI  #+   r@Optional[_AnnotationScanType]c6|duot|tuSrK)r|rrs rA is_pep593rcs   F!25!9!9Y!FFr@rTypeGuard[Iterable[Any]]cpt|tjot|ttf SrK)rmrr r\bytes)rs rAis_non_string_iterablergs6 c?3 4 4 Z c5\>>:r@c.t|tvSrK)r_LITERAL_TYPESrs rArprpms e   ..r@TypeGuard[NewType]c"t|dS)N __supertype__rrs rA is_newtyperqs 5/ * **r@TypeGuard[GenericProtocol[Any]]cBt|dot|dS)Nr8rErrs rAroroys! 5* % % F'%*F*FFr@TypeGuard[TypeAliasType]c,t|tSrK)rmr)rs rA is_pep695r}s e] + ++r@r c^|j}t|r|j}t||SrK)rr)rw super_types rAflatten_newtypers:$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)rwrs rArkrksV%$$t :e,,CCENCCCCCCur@cdSrKr?rs rAde_optionalize_union_typesrs363r@cdSrKr?rs rArrs?Bsr@cdSrKr?rs rArrs #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)rwtyps rArrsw% 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)r4rrzs rAr{z6de_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@)rlrrgroupr splitrjoin)rwrYmmrv 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)rwrrrs rA expand_unionsrsi  %.!!  " KK ! ! !  8eCjj( (:: xr@TypeGuard[ArgsTypeProcotol]c&t|dddS)Nrr UnionType is_origin_ofrs rArrs     r@cLt|ott|vSrK)rr3rrs 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)r|rmtype issubclass)rwrorigins 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)rwrrrs rArrsR u % %F ~u & ! !U * $>&+66v>>r@cttjr|jSt|dd}|t|dd}|S)Nr:_name)rpy310r:r)rwtyp_names rArr"sA |~5*d33  ugt44Hr@c&eZdZd dZd d Zdd Zd S)DescriptorProtoinstanceobjectownerrrHcdSrKr?rLrrs rA__get__zDescriptorProto.__get__.rNr@valuer4cdSrKr?rLrr s rA__set__zDescriptorProto.__set__0rNr@cdSrKr?rLrs rA __delete__zDescriptorProto.__delete__2rNr@N)rrrrrHr)rrr rrHr4rrrHr4)r:r;r<r rrr?r@rArr-s4????====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?r s rAr zDescriptorReference.__get__ErNr@r r4cdSrKr?rs rArzDescriptorReference.__set__GrNr@cdSrKr?rs rArzDescriptorReference.__delete__IrNr@N)rrrrrHr)rrr rrHr4rr:r;r<r=rr rrr?r@rArr8sR9EEEECCCC888888 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?r s rAr zRODescriptorReference.__get__YrNr@r rcdSrKr?rs rArzRODescriptorReference.__set__[rNr@cdSrKr?rs rArz RODescriptorReference.__delete__]rNr@N)rrrrrHr)rrr rrHr)rrrHrrr?r@rArrOsR=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 rrrrrHr!cdSrKr?r s rAr zCallableReference.__get__nrNr@r r4cdSrKr?rs rArzCallableReference.__set__prNr@cdSrKr?rs rArzCallableReference.__delete__rrNr@N)rrrrrHr!)rrr r!rHr4rrr?r@rAr#r#csR9CCCCAAAA888888 99r@r#)rWrXrYrZr[r\r]r^rTr_rUr`rVrarHrX)rwrZrHrZ)rYrrvr7rHrD) rr\rr\r]rrdrrHr)rr\rr\r]rrHr)rr\rr\rHr\) rWrXrYr6r[r\r]r^rTr_rHrX)rwrrHr`)rrrHr)rwrZrHr`)rwrrHr)rwrZrHr)rwrZrHr)rwr rHrX)F)rwrZrr`rHr)rwr\rHr\)rwrXrHrX)rwr rHrZ)rrZrHrX)FF)rwrXrr`rr`rHr)rwrrHr)rwrrHr`)rwrrrrHr`)rwrrr\rrrHr`)rwrXrHr\)e __future__rrcollections.abcabcrrrtypingrrrrr r r r r rrrrrrrrrrtyping_extensionsrrrrrrrr r!r"r#r$r%r&r'r(r)r*r,r-r/r1r2py38 frozensetrrrr3rrrr|r\rZr6rCrG _LiteralStarrgrrtrnrrrrrrprrorrrkrrrrrrrrrrrrrrrr!r#r?r@rAr0se#"""""))))))  A888888<<<<<<100000000000666666::::::444444<<<<<<888888666666@@@@@@888888888888888888......@@@@@@ WT genn T * * * W\ 6 6 6  genn T * * * ;*Y899NNYy))N <+******tDzzH *V  IsJ8NN .....x...hrl(33333Xc6k2333s| ;?!(,DDDDDDN&&&&R////,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@