e4dZddlmZddlZddlZddlZddlmZmZddl m Z m Z ddl Z ddl m Z ddlmZddlmZe e ege j&feege e j&fZ d dd ZGd d ZGd dZGddeej2ZGddeZGddeZGddeZGddeZedej>d f ddZ y)z|Visitor doing some post-processing on the astroid tree. Try to resolve definitions (namespace) dictionary, relationship... ) annotationsN)ABCabstractmethod)CallableOptional)nodes) constants)utilsFc|rtd|d ||S#tjj$r}t|Yd}~yd}~wt$rt j YywxYw)Nzparsing z...)printastroid exceptionsAstroidBuildingException Exception traceback print_exc)funcmodnameverboseexcs )r6idlenr9r(s r__repr__zProject.__repr__Vs/499-tBtH:RDLL8I7J*UUr#N)r5)r6strrD nodes.Moduler+r,)r6rNr+rP)r+zlist[nodes.Module])r+rN) r-r.r/r0r"rErGrIrMr1r#rr3r3=s7 '"!Vr#r3ceZdZdZdddZddZddZddZddZddZ e ddZ dd Z dd Z dd Z dd Zy )Linkera_Walk on the project tree and resolve relationships. According to options the following attributes may be added to visited nodes: * uid, a unique identifier for the node (on astroid.Project, astroid.Module, astroid.Class and astroid.locals_type). Only if the linker has been instantiated with tag=True parameter (False by default). * Function a mapping from locals names to their bounded value, which may be a constant like a string or an integer, or an astroid node (on astroid.Module, astroid.Class and astroid.Function). * instance_attrs_type as locals_type but for klass member attributes (only on astroid.Class) * associations_type as instance_attrs_type but for association relationships * aggregations_type as instance_attrs_type but for aggregations relationships ctj|tjj|||_||_t |_|jjtyr) rr"r LocalsVisitortagprojectAggregationsHandlerassociations_handlerset_nextOtherAssociationsHandler)r rVrUs rr"zLinker.__init__tsU!!$' $$T* $7$9! !!**+C+EFr#c|jr|j|_|jD]}|j |y)zbVisit a pyreverse.utils.Project node. * optionally tag the node with a unique id N)rUr)r7r9visit)r rDmodules r visit_projectzLinker.visit_project~s: 88'')DHll F JJv  r#ct|drytjt|_g|_g|_|jr|j|_ yy)zVisit an astroid.Module node. * set the locals_type mapping * set the depends mapping * optionally tag the node with a unique id locals_typeN) hasattr collections defaultdictlistr`depends type_dependsrUr)r7rCs r visit_modulezLinker.visit_modulesO 4 ' &2248  88'')DH r#ct|drytjt|_|j r|j |_|jdD]'}t|dg}|j|||_ )tjt|_ tjt|_ tjt|_t|j j#D]R}|D]K}t%|t&j(r|j*j-|||j/||MTy)zVisit an astroid.Class node. * set the locals_type and instance_attrs_type mappings * optionally tag the node with a unique id r`NF)recursspecializations)rarbrcrdr`rUr)r7 ancestorsgetattrrBrjinstance_attrs_typeaggregations_typeassociations_typetupleinstance_attrsr= isinstancerUnknownrXhandlehandle_assignattr_type)r rDbaseobjrj assignattrs assignattrs rvisit_classdefzLinker.visit_classdefs 4 ' &2248 88'')DH~~U~3 6G%g/@"EO  " "4 (&5G # 6 $/#:#:4#@ !,!8!8!>!,!8!8!> !4!4!;!;!=> BK) B !*emm<--44ZF// DA B Br#ct|drytjt|_|j r|j |_yy)zVisit an astroid.Function node. * set the locals_type mapping * optionally tag the node with a unique id r`N)rarbrcrdr`rUr)r7rCs rvisit_functiondefzLinker.visit_functiondefs@ 4 ' &2248 88'')DH r#cDt|dryd|_|j|jvr|j}n|j }t|dsit |t jr|j|n=t |t jr|j|n|j||j|j}tt|tj |z|j|j<y)zFVisit an astroid.AssignName node. handle locals_type _handledNTr`)rar}r6framerootrrrClassDefry FunctionDefr{rgr`rdsetr infer_node)r rDr~currents rvisit_assignnamezLinker.visit_assignnames 4 $  99 $JJLEIIKEum,%0##E*E5#4#45&&u-!!%(##DII.'+CL5;K;KD;Q,Q'R$))$r#ct|j|j}t|t j |z|j|j<y)zOHandle an astroid.assignattr node. handle instance_attrs_type N)rrmattrnamerdr r)rDparentrs rruzLinker.handle_assignattr_typesI f00?@48 e&&t, ,5 ""4==1r#c|jj}|jD];}tjj |d|}|j ||d|=y)zKVisit an astroid.Import node. resolve module dependencies rN)rfilenamesr modutils is_relative_imported_module)r rD context_filer6relatives r visit_importzLinker.visit_importsZ yy{'' JJ ;D''33DG\JH  ! !$Q : ;r#c|j}|jj}|!tjj ||}nd}|j D]c}|ddk(r |d|d}|jddkDr! tjj||}||k7sQ|j|||ey#t$rYrwxYw)zOVisit an astroid.ImportFrom node. resolve module dependencies NFr*.) rrrr rrrfindget_module_part ImportErrorr)r rDbasenamerrr6fullnames rvisit_importfromzLinker.visit_importfroms <<yy{''  #''33HlKHHJJ @DAw#~"1T!WI.H}}S!B&&//??,WH8#%%dHh? @#s B?? C  C ctjj|jj}||k(rytj j ||fS)z,Should the module be added to dependencies ?F)osr8dirnamerVr rmodule_in_path)r context_namemod_path package_dirs rcompute_modulezLinker.compute_modulesFggoodll&7&78 8 #..x+HHr#c |j}|j}|r(dj|jdddd|}|j ||r6t |dsg|_|j }||vr|j|yyy)z8Notify an imported module, used to analyze dependencies.rNrre)rr6joinsplitrrarerB)r rDrrr]r mod_pathss rrzLinker._imported_modules{{ ((<#5#5c#:3B#?@A8*MH   |X 669-!#Iy(  *) 7r#NF)rVr3rUboolr+r,)rDr3r+r,rO)rDnodes.ClassDefr+r,)rDznodes.FunctionDefr+r,)rDznodes.AssignNamer+r,rDznodes.AssignAttrrrr+r,)rDz nodes.Importr+r,)rDznodes.ImportFromr+r,)rrNrrNr+r)rDznodes.Import | nodes.ImportFromrrNrrr+r,)r-r.r/r0r"r^rgryr{r staticmethodrurrrrr1r#rrRrRZsr2G *B4 *S<  ;@0I+3+?B+NR+ +r#rRc8eZdZe ddZeddZy)AssociationHandlerInterfacecyrr1r handlers rrYz$AssociationHandlerInterface.set_next#s r#cyrr1r rDrs rrtz"AssociationHandlerInterface.handle)s r#Nrrr+rr)r-r.r/rrYrtr1r#rrr"s8 2 $    r#rc>eZdZUdZded< ddZeddZy) AbstractAssociationHandlera Chain of Responsibility for handling types of association, useful to expand in the future if we want to add more distinct associations. Every link of the chain checks if it's a certain type of association. If no association is found it's set as a generic association in `associations_type`. The default chaining behavior is implemented inside the base handler class. r _next_handlerc||_|Sr)rrs rrYz#AbstractAssociationHandler.set_next<s%r#cV|jr|jj||yyr)rrtrs rrtz!AbstractAssociationHandler.handleBs&       % %dF 3 r#Nrr)r-r.r/r0__annotations__rYrrtr1r#rrr.s8 /.2 $ 44r#rc eZdZdfd ZxZS)rWct|jtjtjfrt|jj t jjr[t|j|j}t|tj|z|j|j<yt|A||yr)rrrr AnnAssignAssignvaluer node_classesNamerrnrrdr rsuperrt)r rDrr __class__s rrtzAggregationsHandler.handleIs dkkEOOU\\#B C KK  w3388I &224==ABG6:%**4007F $ $T]] 3 GN4 (r#r)r-r.r/rt __classcell__)rs@rrWrWHs  ) )r#rWceZdZddZy)rZct|j|j}t|t j |z|j|j<yr)rrorrdr r)r rDrrs rrtzOtherAssociationsHandler.handleVsEf..t}}=>26wAQAQRVAW7W2X  /r#Nr)r-r.r/rtr1r#rrZrZUsYr#rZzno namec`tj}t|}|D]}tjj |s/tj j|jd}nBtjj|r!tjj|d}n|}||j||} | |jxs | j|_|j| | j} | js|j!ddk(s tj j#tjj%| j|D]9}||j||} | | j| k(r)|j| ;|S)z1Return a Project from a list of files or modules.rz __init__.pyr"r)r MANAGERr3rr8existsrfile_from_modpathrisdirr ast_from_filerrEr6packagerget_module_filesr) files func_wrapper project_name black_listrastroid_managerrV somethingfpathast base_names rproject_from_filesr[sPooOl#G( ww~~i($$66ys7KLE WW]]9 %GGLLM:EE?88%I ; ||/sxx 3HH ;;9>>*5; ))::): (#?#@#@%Q;#((i"7""3'  (!(. Nr#r)rzCallable[[str], nodes.Module]rrNrrr+znodes.Module | None) rz list[str]r _WrapperFuncTrrNrztuple[str, ...]rrr+r3)!r0 __future__rrbrrabcrrtypingrrr rpylintr pylint.pyreverser rNModulerrrrr3rTrRrrrWrZDEFAULT_IGNORE_LISTrr1r#rrs>  # #%" seU\\!"C.0FF   '        VV:E+ u22E+P  #  4!<44 )4 )Y9Y#3!"+"?"? " """ "  "  "r#