B S\>@s`dZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddlm Z e Z ee jZeeZeeZddZd d Zd d ZeZefd dZddZee j_ddZddZ d2ddZ!d3ddZ"d4ddZ#ddZ$d5ddZ%d d!Z&d"d#Z'd$d%Z(d&d'Z)d6d(d)Z*d*d+Z+Gd,d-d-Z,iZ-d.d/Z.d0d1Z/e/dS)7zthis module contains a set of functions to create astroid trees from scratch (build_* functions) or from living object (object_build_* functions) N)bases)manager) node_classes)nodescCs.t|dd}|o,t|o,|jdko,|jdkS)N__self___ioio)getattrinspectZismodule__name__ __module__)memberZ member_selfr3lib/python3.7/site-packages/astroid/raw_building.py_io_discrepancy)s    rcCs||_||dS)N)nameadd_local_node)parentnoderrrr_attach_local_node4srcCsH|j}t|dd}|sdSdd|jD}t|||j}|g|jd<dS)zFAdd a __class__ member to the given func node, if we can determine it.r NcSsg|] }|jqSr)r ).0Zancestorrrr ?sz%_add_dunder_class.. __class__)rr __bases__ build_class__doc__instance_attrs)funcr Z python_clsZcls_nameZ cls_basesZ ast_klassrrr_add_dunder_class9s rcCst}||_t|||dS)zdcreate a dummy node and register it in the locals of the given node with the specified name N)r EmptyNodeobjectr)rrZruntime_objectZenoderrrattach_dummy_nodeGsr!cCs|jdk o|jtk S)N)r _marker)selfrrr_has_underlying_objectPsr$cCs ||jkrt|t||dS)zdcreate a Const node and register it in the locals of the given node with the specified name N)Zspecial_attributesrr const_factory)rrvaluerrrattach_const_nodeWs r'cCs"t||dfg}t|||dS)zicreate a ImportFrom node and register it in the locals of the given node with the specified name N)r ImportFromr)rmodnameZ membernameZ from_noderrrattach_import_node_sr*cCs tj||dd}d|_d|_|S)z,create and initialize an astroid Module nodeF)Z pure_pythonN)rZModulepackager)rdocrrrr build_modulegsr-rcCs>t||}x,|D]$}t}||_|j|||_qW|S)z.create and initialize an astroid ClassDef node)rClassDefNamerrappendr)r basenamesr,rbaseZbasenoderrrros    rcCs|pg|p g}}t||}t|_}g|_x4|D],}|jt||jd_||jd_q8Wg|_x*|D]"}|jt |||jd_qtWd|_ d|_ ||_|rt ||S)z1create and initialize an astroid FunctionDef nodeN) rZ FunctionDefZ Argumentsargsr0r/rrdefaultsr%kwargvarargregister_arguments)rr4r5r,rZargsnodeargdefaultrrrbuild_functionzs$    r;cCst|dd|DS)z.)rr()Zfromnamenamesrrrbuild_from_importsr=cCs~|dkrD|jj}|jjr*||jj|j|jjrD||jj|jx4|D],}t|tjrj||j|qJt||j qJWdS)z{add given arguments to local args is a list that may contains nested lists (i.e. def func(a, (b, c, d)): ...) N) r4r7Z set_localr6 isinstancerr/rr8Zelts)rr4r9rrrr8s  r8cCs dd|jD}t||||dS)z(create astroid for a living class objectcSsg|] }|jqSr)r )rr2rrrrsz&object_build_class..) localname)r_base_class_object_build)rr r?r1rrrobject_build_classsrAcCsbt|\}}}}|dk r$|||dk r6||tt|ddpF||||j}|||dS)z+create astroid for a living function objectNr )r Z getargspecr0r;r rr)rr r?r4ZvarargsZvarkwr5rrrrobject_build_functions  rBcCst||g|S)z2create astroid for a living data descriptor object)r@)rr rrrrobject_build_datadescriptorsrCcCs<tt|ddp||jd}d|j_|||t||dS)z4create astroid for a living method descriptor objectr N)r,)r;r rr4rr)rr r?rrrrobject_build_methoddescriptors  rDc Cst|pt|ddp|||j}t|t|_|||yt|trL|j }nt Wn Yn>Xx:| D].\}}t } || _|| _d| _| g|j|<qjW|S)zbcreate astroid for a living class object, with a given set of base names (e.g. ancestors) r N)rr rr>typeZ _newstyler issubclass Exception__dict__ TypeErroritemsrrr rlinenor) rr r1rr?klassZinstdictZ item_nameobjZvalnoderrrr@s&    r@cCsy |j}Wntk r"d}YnXt|dd}|dkrTt|tsFtt|||n*|t|ddkrrt|||n t|||dS)N co_filename__file__) __code__AttributeErrorr r>r AssertionErrorrDr!rB)rrr modulecodefilenamerrr_build_from_functions   rWc@s2eZdZdZddZd ddZddZd d ZdS) InspectBuilderzclass for building nodes from living object this is actually a really minimal representation, including only Module, FunctionDef and ClassDef nodes and some others as guessed. cCsi|_d|_dS)N)_done_module)r#rrr__init__szInspectBuilder.__init__NcCs||_|dkr|j}yt||j}Wntk r@t|}YnX|rRtj|n||_|_||_ t |t |d|_ i|_||||S)zbuild astroid from a living module (i.e. using inspect) this is used when there is no python source code available (either because it's a built-in module or because the .py is not available) N__path__)rZr r-rrRospathabspathfilerMANAGER cache_modulehasattrr+rY object_build)r#rTr)r^rrrr inspect_build s   zInspectBuilder.inspect_buildc Cs||jkr|j|S||j|<xt|D]}yt||}Wn tk r^t||w*YnXt|rp|j}t|rt ||||j q*t |rt |s| |||rq*t|||q*t|rN| |||rq*||jkr|j|}||j|dkr(|||nt|||}||||dkr|jdkr|j|j |_q*t|rxt|tsjtt|||q*t|rt|tstt|||q*t|trt|||q*t|rt ||||j q*t|||q*WdS)z~recursive method which create a partial ast from real objects (only function, class, and method are handled) rrN)rYdirr rRr!r Zismethod__func__Z isfunctionrWrZZ isbuiltinrimported_memberrDZisclasslocalsgetrrArdrZismethoddescriptorr>r rSZisdatadescriptorrC _CONSTANTSr'Z isroutine)r#rrNrr Z class_noderrrrd!sR                  zInspectBuilder.object_buildc Csyt|dd}Wntdd}YnX|dkrT|dkrDtj}nt|||dSddd||}||jjkryttj ||Wn$t t fk rt|||YnXt |||dSd S) z1verify this is not an imported class or handle itr Nz:unexpected error while building astroid from living object)__new____subclasshook__TZgtk_gtkr)ZgtkrF) r _LOGZ exceptionbuiltinsr r!rjrZsysmodulesKeyErrorrRr*)r#rr rr)Z real_namerrrrhWs(    zInspectBuilder.imported_member)NN)r r __qualname__rr[rerdrhrrrrrXs  6rXcCs t|jjS)N) _CONST_PROXYr&r)Zconstrrr _set_proxiedsruc Cst}|t}x~tjD]p\}}|tdkr@td}||_n,|tt kr\td}||_n| |j d}|t t ttfkr||_q|t|<qWtttj_ttjj tjj}||_|tj_|tjjtjttdrttjj tjj}||_|tj_|tjjtjtj tjtj!tdtt tj"tj#tj$tj%tj&f }xF|D]>}|j |krFt|j |j}||_||||||j <qFWdS)z)astroid bootstrapping the builtins moduleNZNoneTypeZNotImplementedTyperAsyncGeneratorType)'rXreror CONST_CLSrKrFrrNotImplementedr r dictlistsettupleZ_proxiedrtpropertyrurZConstr.types GeneratorTyperr GeneratorrdrcrvAsyncGeneratorGetSetDescriptorTypeMemberDescriptorType FunctionType MethodTypeBuiltinFunctionType ModuleType TracebackType) ZbuilderZastroid_builtinclsZnode_clsproxyZ_GeneratorTypeZ_AsyncGeneratorTypeZ builtin_typesZ_typerrr_astroid_bootstrappingsR         r)N)rN)NNN)N)NN)0rror Zloggingr]rpr~ZastroidrrrrZAstroidManagerrar|rwrkvarsZ _BUILTINSZ getLoggerr rnrrrr r"r!r$rZhas_underlying_objectr'r*r-rr;r=r8rArBrCrDr@rWrXrtrurrrrrsN             }<