c'8DUdZddlmZddlZddlZddlmZddlmZddl m Z m Z m Z ddl mZddlmZmZdd lmZmZmZdd lmZdd lmZmZmZmZmZmZmZm Z m!Z!m"Z"dd l#m$Z$m%Z%dd l&m'Z'ej(dkrddlm)Z)nddl*m)Z)ddhZ+hdZ,de-d<ddhZ.de-d<dZ/e0e1edgZ2e3dZ4dZ5d7dZ6d8dZ7d7d Z8 d8d9d&Z9d:d(Z: d8d;d*Z; d8dd0Z> d8d?d1Z?d=d2Z@ d8d?d3ZAd=d4ZB d8d@d6ZCeDee e7e6eDe!e e9e8eDee eCeBer)eDe%e e<e:n*er(eDe$e e;e:eDee e?e=eDee eAe@dS)Az$Astroid hooks for typing.py support.) annotationsN)Iterator)partial)context extract_node inference_tip)_extract_single_node) PY38_PLUS PY39_PLUS)AttributeInferenceErrorInferenceErrorUseInferenceDefault)AstroidManager) Assign AssignName AttributeCallConst JoinedStrNameNodeNG SubscriptTuple)ClassDef FunctionDef) Uninferable))FinalTypeVarNewType>typing.NewTypetyping.TypeVartyping_extensions.TypeVarrTYPING_TYPEVARS_QUALIFIEDztyping.TypedDictztyping_extensions.TypedDictTYPING_TYPEDDICT_QUALIFIEDz class Meta(type): def __getitem__(self, item): return self @property def __args__(self): return () class {0}(metaclass=Meta): pass __all__)(ztyping.Hashableztyping.Awaitableztyping.Coroutineztyping.AsyncIterableztyping.AsyncIteratorztyping.Iterableztyping.Iteratorztyping.Reversiblez typing.Sizedztyping.Containerztyping.Collectionztyping.Callableztyping.AbstractSetztyping.MutableSetztyping.Mappingztyping.MutableMappingztyping.Sequenceztyping.MutableSequenceztyping.ByteStringz typing.Tuplez typing.Listz typing.Dequez typing.Setztyping.FrozenSetztyping.MappingViewztyping.KeysViewztyping.ItemsViewztyping.ValuesViewztyping.ContextManagerztyping.AsyncContextManagerz typing.Dictztyping.DefaultDictztyping.OrderedDictztyping.Counterztyping.ChainMapztyping.Generatorztyping.AsyncGeneratorz typing.Typeztyping.Patternz typing.Matchz? @classmethod def __class_getitem__(cls, item): return cls returnboolc|j}t|tr|jtvSt|t r|jtvSdS)NF)func isinstancerattrnameTYPING_TYPEVARSrname)noder+s :lib/python3.11/site-packages/astroid/brain/brain_typing.py$looks_like_typing_typevar_or_newtyper2tsL 9D$ ""0}//$,yO++ 5c$ t|j|}n!#ttf$r }t |d}~wwxYw|tvrt |jst t|jdtrt |jd d}tt|}||S)z8Infer a typing.TypeVar(...) or typing.NewType(...) call.rNr')nextr+inferr StopIterationrqnamer%argsr,r as_stringstriprTYPING_TYPE_TEMPLATEformat)r0 context_ittonr+exctypenames r1infer_typing_typevar_or_newtyperC}s+DIOOMO::;; M *+++!s*+ zz||44"!! 9"!!$)A, **"!!y|%%''--c22H ,33H== > >D ::m: , ,,s(+A AA ct|tr|jtvSt|tr|jtvSt|t rt|jSdS)zLTry to figure out if a Subscript node *might* be a typing-related subscript.F) r,rr/TYPING_MEMBERSrr-r_looks_like_typing_subscriptvaluer0s r1rFrFsg$+yN**$ ""/}..$ ""8+DJ777 5r3r0rctxcontext.InferenceContext | NoneIterator[ClassDef]c t|j}n!#ttf$r }t |d}~wwxYw|dr|tvrt t|tr|dvrtt}|g|j d<t|jtr\||jjvrNt!|jddr8|jj}||jj ||jj=t)|gSt+t,|dd}||S) z Infer a typing.X[...] subscript.Nztyping.>typing.Generictyping.Annotatedtyping_extensions.Annotated__class_getitem____cache.r5)r7rGr8r r9rr: startswith TYPING_ALIASr,rr CLASS_GETITEM_TEMPLATElocalsparentbasesgetattrrQgetslotsiterrr>r?split)r0rIrGrA func_to_addcaches r1infer_typing_attrras+TZ%%''(( M *+++!s*+ ;;== # #I . ."%++--<2O"!!%""u{{}}9(++ABB -8M () t{H - - - )) - Y55 -K'Eyy*++ -$++,UG}} ,33EKKMM4G4G4L4LR4PQQ R RD ::c: " ""s&)AAAFunctionDef | ClassDefc8|tvS)z'Check if node is TypedDict FunctionDef.)r:r&rHs r1_looks_like_typedDictrds ::<<5 55r3rcVtd}|g|jd<t|gS)Ndict__call__)r rWr])r0rIr_s r1infer_old_typedDictrhs-'v..K*mDK  <<r3rctd|j|j|j}|t dggdt d}|g|jd<t|gS)z,Replace TypedDict FunctionDef with ClassDef. TypedDictr/lineno col_offsetrXrfNrYbody decoratorsrg) rrlrmrXpostinitrr rWr])r0rI class_defr_s r1infer_typedDictrss~ {?{ I l6223"NNN&v..K$/=IZ   r3rct|jto6|jjdko&t|jdt tfS)z Returns True if the node corresponds to a call to _alias function. For example : MutableSet = _alias(collections.abc.MutableSet, T) :param node: call node _aliasr)r,r+rr/r;rrHs r1_looks_like_typing_aliasrvsH 49d## INh &  ty|i%6 7 7 r3Nonecd} |dt||j}||_dS#t$rYdSwxYw)zJDisable the access to __class_getitem__ method for the node in parameters.cD|dkrtd||g|Ri|S)z Raises an AttributeInferenceError in case of access to __class_getitem__ method. Otherwise, just call origin_func. rPz'__class_getitem__ access is not allowed)r ) origin_funcattrr;kwargss r1 full_raiserz1_forbid_class_getitem_access..full_raisersB & & U)*STT T{41$111&111r3rPN)rZrr )r0r}partial_raisers r1_forbid_class_getitem_accessrsi222   ())) !dl;;% "     s18 AAct|jtrBt|jjdkr%t|jjdt st  t|jd |}n/#t$r"}t|jd||d}~wwxYw|jjd}t|j |j|j|j}|t kr.t|tr||ggd|jd}t$st|t&r.|js't$r@t|t*r+|jdkr t/t0}|g|jd<nt5|t7|gS) z Infers the call to _alias function Insert ClassDef, with same name as aliased class, in mro to simulate _GenericAlias. :param node: call node :param context: inference context rr5r0rNrkrnrP)r,rXrlentargetsrrr7r;r8r9r rr/rlrmrrqr reltsrrGr rVrWrr])r0rIrese assign_namerrmaybe_type_varr_s r1infer_typing_aliasrs t{F + +"4;&''1,"$+-a0*==" "!D49Q<%%c%2233 DDD$)A,<<>K-8MI()   rct|toSt|jtr|jjdkp)t|jt o|jjdkS)Ncast)r,rr+rr/rr-rHs r1_looks_like_typing_castrs_ dD ! ! 49d## % INf $ ) di + + ) I & ( r3Iterator[NodeNG]ct|jttfst t |j|}n!#ttf$r }t|d}~wwxYwt|tr0| dkst|j dkrt|j d|S)z>Q " "! 9Q<  c  * **s(AA1$A,,A1)r(r))N)r0rrIrJr(rK)r0rbr(r))r0rrIrJr(rK)r0rrIrJr(rK)r0rr(r))r0rr(rw)r0rrIrJr(rK)r0rrIrJr(r)E__doc__ __future__rsystypingcollections.abcr functoolsrastroidrrrastroid.builderr astroid.constr r astroid.exceptionsr r rastroid.managerrastroid.nodes.node_classesrrrrrrrrrrastroid.nodes.scoped_nodesrr astroid.utilr version_infortyping_extensionsr.r%__annotations__r&r>setrZrE frozensetrUrVr2rCrFrardrhrsrvrrrrrrregister_transformr3r1rs= +**"""""" $$$$$$8888888888000000........ +*****                        =<<<<<<<$$$$$$v(''''''i($$$ &89V$WWWWW WWVY3344y)++ Z----(=A$#$#$#$#$#N6666<@?C      (    28<33333lD8<48<+++++*##M122( ## }}.//1M##--) * *,C N'']]?335JN''-- 3446K##--* + +-E##--+ , ,.Gr3