e ?:UdZddlmZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z ddlmZdd lmZmZdd lmZmZdd lmZmZmZmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)m*Z*ddhZ+hdZ,de-d<ddhZ.de-d<dZ/e0e1edgZ2e3dZ4dZ5d+dZ6 d, d-dZ7d+dZ8 d, d.dZ9 d/dZ: d, d0d Z; d, d1d!Z d, d4d$Z?d2d%Z@ d, d4d&ZAd2d'ZB d, d5d(ZCd)ZDd6d*ZEy)7z$Astroid hooks for typing.py support.) annotationsN)Iterator)partial)Final)context extract_node inference_tip)register_module_extender)AstroidBuilder_extract_single_node) PY39_PLUS PY312_PLUS)AstroidSyntaxErrorAttributeInferenceErrorInferenceErrorUseInferenceDefault)AstroidManager) Assign AssignName AttributeCallConst JoinedStrNameNodeNG SubscriptTuple)ClassDef FunctionDefTypeVarNewType>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 c|j}t|tr|jtvSt|t r|j tvSy)NF)func isinstancerattrnameTYPING_TYPEVARSrname)noder)s typing.Generictyping.Annotatedtyping_extensions.Annotated__class_getitem____cachectgSNiter)r.rrEs r/z#infer_typing_attr..s ugr1.r3)r5rEr6rr7rr8 startswith TYPING_ALIASr*rr CLASS_GETITEM_TEMPLATElocalsparentbasesgetattrrLgetslots_explicit_inferencerPrr<r=split)r.ctxr? func_to_addcacherEs @r/infer_typing_attrrbs_+TZZ%%'( ;;= # #I .%++-<2O!!%"u{{}9(++AB -8M () t{{H - ))) Y5KK''Eyy**+7$++++,#F UG} ,33EKKM4G4G4LR4PQ RD ::c: ""A M *+!s*+s#F##F>2F99F>c.|jtvS)z'Check if node is TypedDict FunctionDef.)r8r&rFs r/_looks_like_typedDictrds ::<5 55r1cPtd}|g|jd<t|gS)Ndict__call__)r rWrP)r.r_r`s r/infer_old_typedDictrhs*'v.K*mDKK  <r1ctd|j|j|j|j|j }|j tdggdtd}|g|jd<t|gS)z,Replace TypedDict FunctionDef with ClassDef. TypedDictr-lineno col_offsetrX end_linenoend_col_offsetrfNrYbody decoratorsrg) rrlrmrXrnropostinitrr rWrP)r.r_ class_defr`s r/infer_typedDictrus~ {{??{{??** Il623"N&v.K$/=IZ  r1ct|jtxr=|jjdvxr#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 >_alias_DeprecatedGenericAliasr)r*r)rr-r9rrFs r/_looks_like_typing_aliasrysK 499d# IINNC C tyy|i%6 7 r1cd} |jdt||j}||_y#t$rYywxYw)zJDisable the access to __class_getitem__ method for the node in parameters.c:|dk(r td||g|i|S)z Raises an AttributeInferenceError in case of access to __class_getitem__ method. Otherwise, just call origin_func. rKz'__class_getitem__ access is not allowed)r) origin_funcattrr9kwargss r/ full_raiserz1_forbid_class_getitem_access..full_raisers. & &)*ST T41$1&11r1rKN)rZrr)r.rpartial_raisers r/_forbid_class_getitem_accessrsD2   () !dll;% "   s.4 AAct|jtrIt|jjdk(r't|jjdt st  t|jdj|}|jjd}t|j|j|j|j|j |j"t|trj%|ggd|jd}t&st|t(r1|j*s%t&r?t|t,r/|j.dkDr t1t2}|gj4d<n t7fd |_t;gS#t$r }t|jd||d}~wwxYw) a  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 # TODO: evaluate if still necessary when Py3.12 is minimum rr3r.rNrkrprKctgSrNrOr.rrts r/rQz$infer_typing_alias..NT9+5Fr1)r*rXrlentargetsrrr5r9r6r7rrr-rlrmrnrorsr reltsrrEr rVrWrr]rP)r.r_rese assign_namemaybe_type_varr`rts @r/infer_typing_aliasrs t{{F +4;;&&'1,$++--a0*=!!D499Q<%%c%23++%%a(K   !!)){{))"11 I#x  #RDAYYq\N NE2>;N;N  ~u -  1 $++AB 1< ,- %Y/ GD  K D$))A,.rr1)r*rXrrrrrr5r9r6r7rrr-rlrmrnrorsr rVrWr]rP)r.r_rrrr`rts @r/infer_special_aliasrss: 4;;'  ## $ ) t{{**1-z :!!D499Q<%%c%23++%%a(K   {{!!))))"11 IcU=&'=>K-8MI()FD  # D$))A,Q !! 99Q<  c  ** M *+!s*+s%B--C<CCclttjtjdS)Na class Generic: @classmethod def __class_getitem__(cls, item): return cls class ParamSpec: ... class ParamSpecArgs: ... class ParamSpecKwargs: ... class TypeAlias: ... class Type: @classmethod def __class_getitem__(cls, item): return cls class TypeVar: @classmethod def __class_getitem__(cls, item): return cls class TypeVarTuple: ... )r r string_buildtextwrapdedentr1r/_typing_transformrs. .* + 8 8   r1cr|jtttt|jt tt t|jtttttr)|jttttn(|jtttt|jttt t"|jttt$t&t(rt+|dt,yy)Ntyping)register_transformrr rAr0rrbrDrrr rrurdrrhrryrrrr r)managers r/registerrs  56,  =!235Q  m-.0G"" 79N  "" m$78:O   m./1I  m/02K (4EFr1)returnboolrN)r.rr>context.InferenceContext | NonerIterator[ClassDef])r.rr_rrr)r.zFunctionDef | ClassDefrr)r.rr_rrr)r.rr_rrr)r.rrr)r.rrNone)r.rr_rrr)r.rr_rrzIterator[NodeNG])rrrr)F__doc__ __future__rrrcollections.abcr functoolsrrastroidrrr astroid.brain.helpersr astroid.builderr r astroid.constr rastroid.exceptionsrrrrastroid.managerrastroid.nodes.node_classesrrrrrrrrrrastroid.nodes.scoped_nodesrrr,r%__annotations__r&r<setrZrC frozensetrUrVr0rArDrbrdrhruryrrrrrrrrrr1r/rs +" $88:@/ +   =i($5 &89V$WEW WVY34)+ ZBF- ->--2=A&# &#9&#&#R6 6 6<@ 8?C ;$* 28<: :4::zD8< 4@8<+ +4++*.Gr1