B aR @s4dZddlZddlmZddlmZmZddlmZm Z ddl m Z m Z m Z mZedZedZed Zd Zed d hZe red dddddddddddh Zd-ddZGdddZd.ddZdd ZGd!d"d"eZGd#d$d$eZGd%d&d&eZGd'd(d(eZGd)d*d*eZGd+d,d,eZ dS)/zXThis module contains base classes and functions for the nodes and some inference utils. N)context) decoratorsutil)BUILTINS PY310_PLUS)AstroidTypeErrorAttributeInferenceErrorInferenceErrorNameInferenceErrorzinterpreter.objectmodelhelpersmanager__bool__z .propertyzabc.abstractpropertyz enum.propertycached_propertyZcachedpropertyZ lazypropertyZ lazy_propertyZreifyZ lazyattributeZlazy_attributeZ LazyPropertyZlazyZcache_readonlyDynamicClassAttributecs|j|d}t|rdSdd|DtfddtDrBdS|jsLdSx|jjpXdD]x}tj||d}|dksZ|t j krqZ|j j d krZxD|j D]:}|j j d krq||j\}}|jtkr|jd krdSqWqZWdS) N)rTcSs$h|]}|tjk r|ddqS).)r Uninferablesplit).0namer1/tmp/pip-unpacked-wheel-mm06h1t3/astroid/bases.py Osz_is_property..c3s|]}|kVqdS)Nr)rr)strippedrr Rsz_is_property..FrClassDefNameproperty)decoratornames PROPERTIES intersectionanyPOSSIBLE_PROPERTIESrZnodesr Z safe_inferrr __class____name__baseslookuprr)methrr decoratorinferredZ base_classmodule_r)rr _is_propertyKs*      r,c@s0eZdZdZdZd ddZddZd ddZdS) Proxyza simple proxy object Note: Subclasses of this object will need a custom __getattr__ if new instance attributes are created. See the Const class NcCs|dk r||_dS)N)_proxied)selfZproxiedrrr__init__rszProxy.__init__cCs0|dkr|jjS||jkr$|j|St|j|S)Nr.)r#r.__dict__getattr)r/rrrr __getattr__vs   zProxy.__getattr__ccs |VdS)Nr)r/rrrrinfer}sz Proxy.infer)N)N)r$ __module__ __qualname____doc__r.r0r3r4rrrrr-gs  r-c csd}|dk r|j}|}n d}t}x|D]}|tjkrH|Vd}q.||||_y$x|j|dD]}|Vd}qfWWq.tk rw.Yq.t k rtjVd}Yq.Xq.W|st d|||ddS)zGReturn an iterator on statements inferred by each statement in *stmts*.FNT)rz.Inference failed for all members of {stmts!r}.)stmtsframer) lookupnameclone contextmodInferenceContextrrZ _infer_namer4r r )r8rr9r)rZstmtrrr _infer_stmtss6    r>c Cst|j||dd}|rt|dr|s0tjSytxn|j||dD]\}|tjkrT|Syt|j|d}Wn.tk r}zt |d|Wdd}~XYnX| SWWnt k rYnXtjS)N)rinfer_call_result) nextigetattrhasattrcallablerrr?r4 StopIterationr bool_value)instance method_namerr'valuer)errr_infer_method_result_truths  rJc@sDeZdZdZdZddZdddZddd Zdd d Zdd d Z dS) BaseInstancezNAn instance base class, which provides lookup methods for potential instances.NcCsdS)Nz Instance ofr)r/rrr display_typeszBaseInstance.display_typeTc Csy|j||}Wnftk rx}zH|jrB||jkrB|j|gS|rX|jj||ddSt|||d|Wdd}~XYnX|ry||jj||ddStk rYnX|S)NF) class_context)target attributer)r.Z instance_attrrspecial_attributesr&r2)r/rr lookupclassvaluesexcrrrr2s" zBaseInstance.getattrc cs|s t}yP||_||jr.td||d|j||dd}t|||||dEdHWnt k ry8|jj j dkr~|jj ||dd}|||EdHWn2t k r}ztft ||Wdd}~XYnXYnXdS) zinferred getattrz%Cannot infer the same attribute again)messagenoderF)rQ)r9Nr)rM)r<r=r:pushr.r r2r> _wrap_attrrr#r$rAvars)r/rrZget_attrattrserrorrrrrAs( zBaseInstance.igetattrccsx|D]}t|tr>t|r0|||EdHqt||Vqt|dr|jdkr|jjrz|jjdjdkrzt||Vq|Vq|VqWdS)z7wrap bound methods of attrs in a InstanceMethod proxiesNrzrr/) isinstance UnboundMethodr,r? BoundMethodrBrargs arguments)r/rYrattrrrrrWs   zBaseInstance._wrap_attrccsrt||}d}xJ|jd|D]8}|tjks |s8q x|||D]}d}|VqFWq W|snt|||ddS)z4infer what a class instance is returning when calledF__call__T)rUcallerrN) r<bind_context_to_noder.rArrrCr?r )r/rbrr)rUresrrrr? s zBaseInstance.infer_call_result)NT)N)N)N) r$r5r6r7rPrLr2rArWr?rrrrrKs   rKc@sZeZdZdZeddZddZddZdd Z d d Z d d Z dddZ dddZ dS)Instancez-A special node representing a class instance.cCstS)N) objectmodelZ InstanceModelrrrrzInstance.cCsd|jj|jjt|S)Nz)formatr.rootrid)r/rrr__repr__szInstance.__repr__cCsd|jjd|jjS)Nz Instance of r)r.rjr)r/rrr__str__"szInstance.__str__cCs.y|jjddddStk r(dSXdS)NraF)rMT)r.r2r)r/rrrrC%s zInstance.callablecCs |jS)N)r.qname)r/rrrpytype,szInstance.pytypecCsdS)Nz Instance ofr)r/rrrrL/szInstance.display_typeNc Csx|p t}tjgd|_||_yt|t|}WnBttfk rryt|d|}Wnttfk rldSXYnX|S)aVInfer the truth value for an Instance The truth value of an instance is determined by these conditions: * if it implements __bool__ on Python 3 or __nonzero__ on Python 2, then its bool value will be determined by calling this special method and checking its result. * when this method is not defined, __len__() is called, if it is defined, and the object is considered true if its result is nonzero. If a class defines neither __len__() nor __bool__(), all its instances are considered true. )r^__len__T) r<r= CallContext callcontextZ boundnoderJBOOL_SPECIAL_METHODr r)r/rresultrrrrE2s  zInstance.bool_valuecCst||}|s|}tj|gd|_t|jd|dd}t|tsPtd||dt |j j dkrnt d||dt| ||dS)N)r^ __getitem__)rz(Could not find __getitem__ for {node!r}.)rUrz8__getitem__ for {node!r} does not have correct signature)r<rcrqrrr@rAr[r]r lenr^r_rr?)r/indexrZ new_contextmethodrrrgetitemMs   zInstance.getitem)N)N)r$r5r6r7rlazy_descriptorrPrlrmrCrorLrErzrrrrres rec@s\eZdZdZeddZddZddZdd Z dd d Z dd dZ ddZ dddZ d S)r\z=a special node representing a method not bound to an instancecCstS)N)rfZUnboundMethodModelrrrrrgfrhzUnboundMethod.cCs,|jj}d|jj|jj|t|S)Nz<{} {} of {} at 0x{}) r.parentr9rir#r$rrnrk)r/r9rrrrlhs zUnboundMethod.__repr__cCsdS)Nrr)r/rrrimplicit_parametersnsz!UnboundMethod.implicit_parameterscCsdS)NFr)r/rrris_boundqszUnboundMethod.is_boundNcCs&||jkr|j|gS|j||S)N)rPr&r.r2)r/rrrrrr2ts zUnboundMethod.getattrcCs*||jkrt|j|fS|j||S)N)rPiterr&r.rA)r/rrrrrrAys zUnboundMethod.igetattrcCsp|jjdkrb|jjdtkrb|jrP|j|jd}|jdj |d}ng}dd|DS|j ||S)a The boundnode of the regular context with a function called on ``object.__new__`` will be of type ``object``, which is incorrect for the argument in general. If no context is given the ``object.__new__`` call argument will correctly inferred except when inside a call that requires the additional context (such as a classmethod) of the boundnode to determine which class the method was called from __new__z %s.objectr)rcss$|]}|tjk rt|n|VqdS)N)rrre)rxrrrrsz2UnboundMethod.infer_call_result..) r.rr|r9rnrr^Z extra_contextgetr4r?)r/rbrZ node_contextr4rrrr?~s zUnboundMethod.infer_call_resultcCsdS)NTr)r/rrrrrEszUnboundMethod.bool_value)N)N)N)r$r5r6r7rr{rPrlr}r~r2rAr?rErrrrr\bs  r\csZeZdZdZeddZddZddZdd Z d d Z dfd d Z dddZ Z S)r]z9a special node representing a method bound to an instancecCstS)N)rfZBoundMethodModelrrrrrgrhzBoundMethod.cCst||||_dS)N)r\r0bound)r/proxyrrrrr0s zBoundMethod.__init__cCs|jdkrdSdS)Nrr)r)r/rrrr}s zBoundMethod.implicit_parameterscCsdS)NTr)r/rrrr~szBoundMethod.is_boundc sddlm}yt|jdjd}Wn.tk rT}ztd|Wdd}~XYnX|jjdkrfdS| dt sxdSyt|jdjd}Wn.tk r}ztd|Wdd}~XYnX|jjdkrdSt |j t sdSyt|jd jd}Wn0tk r,}ztd|Wdd}~XYnX|jjd kr@dSyfd d |jD}Wn0tk r}ztd|Wdd}~XYnXtd d|DrdSyt|jdjd} Wn0tk r}ztd|Wdd}~XYnX| jjdkrdStt} x| jD]\} } yt| jd} Wn0tk r^}ztd|Wdd}~XYnXyt| jd} Wn0tk r}ztd|Wdd}~XYnX| jjdkrt | j t r| | j | qW|j|j |j|j|d} |}| j|j|ggd|gd| | _| S)zTry to infer what type.__new__(mcs, name, bases, attrs) returns. In order for such call to be valid, the metaclass needs to be a subtype of ``type``, the name needs to be a string, the bases needs to be a tuple of classes r) node_classes)rNrz%s.typerZConstrvTuplecsg|]}t|jdqS))r)r@r4)relt)rrr sz4BoundMethod._infer_type_new_call..css|]}|jjdkVqdS)rN)r#r$)rbaserrrrsz3BoundMethod._infer_type_new_call..Dict)rlineno col_offsetr|T)r%bodyrZnewstyle metaclasskeywords)astroidrr@r^r4rDr r#r$Z is_subtype_ofrr[rHstrZeltsr! collections defaultdictlistitemsappendrrZPassZpostinitlocals)r/rbrrZmcsrIrr%Zinferred_basesrYZ cls_localskeyrHclsemptyr)rr_infer_type_new_calls|     z BoundMethod._infer_type_new_callNcsht||j}|jjjdkrZ|jjdkrZ|jdkrZt|jdkrZ|||}|rZt |fSt ||S)Nrtyper) r<rcrr#r$rrwr^rrsuperr?)r/rbrZnew_cls)r#rrr? s    zBoundMethod.infer_call_resultcCsdS)NTr)r/rrrrrEszBoundMethod.bool_value)N)N)r$r5r6r7rr{rPr0r}r~rr?rE __classcell__rr)r#rr]s[r]csneZdZdZeejZdfdd Z e j ddZ ddZ d d Zd d Zdd dZddZddZZS) Generatorzea special node representing a generator. Proxied class is set once for all in raw_building. Ncs t||_t||_dS)N)rr0r|r<Z copy_context _call_context)r/r|Zgenerator_initial_context)r#rrr0$s zGenerator.__init__ccs|j|jEdHdS)N)r|Zinfer_yield_resultr)r/rrrinfer_yield_types)szGenerator.infer_yield_typescCsdS)NFr)r/rrrrC-szGenerator.callablecCsdtS)Nz %s.generator)r)r/rrrro0szGenerator.pytypecCsdS)Nrr)r/rrrrL3szGenerator.display_typecCsdS)NTr)r/rrrrrE6szGenerator.bool_valuecCsd|jj|jt|S)Nz)rir.rrrk)r/rrrrl9szGenerator.__repr__cCs d|jjS)Nz Generator(%s))r.r)r/rrrrm>szGenerator.__str__)NN)N)r$r5r6r7rr{rfZGeneratorModelrPr0rcachedrrCrorLrErlrmrrr)r#rrs  rc@s0eZdZdZddZddZddZdd Zd S) AsyncGeneratorz,Special node representing an async generatorcCsdtS)Nz%s.async_generator)r)r/rrrroEszAsyncGenerator.pytypecCsdS)Nrr)r/rrrrLHszAsyncGenerator.display_typecCsd|jj|jt|S)Nz!)rir.rrrk)r/rrrrlKszAsyncGenerator.__repr__cCs d|jjS)NzAsyncGenerator(%s))r.r)r/rrrrmPszAsyncGenerator.__str__N)r$r5r6r7rorLrlrmrrrrrBs r)N)N)!r7rrrr<rrZ astroid.constrrZastroid.exceptionsrrr r Z lazy_importrfr r rsraddr"r,r-r>rJrKrer\r]rrrrrrsD        "_K9&