B S\1@s^dZddlZddlZddlmZddlmZddlmZddlmZddZ Gd d d Z dS) zastroid manager: avoid multiple astroid build of a same module when possible by providing a class responsible to get astroid representation from various source and using a cache of built modules) N) exceptions)spec)modutils) transformscCs"yt|Stk rdSXdS)Nz???)repr Exception)objr .lib/python3.7/site-packages/astroid/manager.py safe_reprsr c@seZdZdZdZiZddZeddZddZ d(d d Z ddZ ddZ ddZ d)ddZddZddZd*ddZd+ddZd,ddZd d!Zd"d#Zd$d%Zd&d'Zd S)-AstroidManagerzmthe astroid manager, responsible to build astroid from files or modules. Use the Borg pattern. zastroid loadercCs\tj|_|jsXi|_i|_g|_d|_d|_t|_ t |_ |j j |_ |j j|_d|_dS)NFd)r brain__dict__ astroid_cache_mod_file_cache_failed_import_hooksalways_load_extensionsZ optimize_astsetextension_package_whitelistrZTransformVisitor _transformZregister_transformZunregister_transformZmax_inferable_values)selfr r r __init__.s   zAstroidManager.__init__cCs |jdS)Nbuiltins)r)rr r r builtins_module?szAstroidManager.builtins_modulecCs |j|S)z8Visit the transforms and apply them to the given *node*.)rZvisit)rZnoder r r visit_transformsCszAstroidManager.visit_transformsNTFcCsytj|dd}d}Wntjk r,YnX|dkrdydt|}Wntk rb|}YnX||jkr|j|j|kr|j|S|rddlm }|| ||S|r|r| |St j d|ddS) z.given a module name, return the astroid objectT)Zinclude_no_extN.r)AstroidBuilderz"Unable to build an AST for {path}.)path)rZget_source_fileZ NoSourceFilejoinZmodpath_from_file ImportErrorrfileastroid.builderrZ file_buildast_from_module_namerAstroidBuildingError)rfilepathmodnamefallbacksourcerr r r ast_from_fileGs(     zAstroidManager.ast_from_filecCsddlm}||d|S)Nr)r)r"r string_build)rr&rr r r _build_stub_modulebs z!AstroidManager._build_stub_modulecCsddlm}|||S)Nr)build_namespace_package_module)r"r-)rr&rr-r r r _build_namespace_modulegs z&AstroidManager._build_namespace_modulecsHjr dSt|rdS|dtfddtdtdDS)NTrc3s&|]}dd|jkVqdS)rN)rr).0x)partsrr r ssz5AstroidManager._can_load_extension..)rrZis_standard_modulesplitanyrangelen)rr&r )r1rr _can_load_extensionls   z"AstroidManager._can_load_extensionc Cs||jkr|j|S|dkr&||St}|rDttj|zy.|||}|jt j j kr| |j }|dk r~|Sn|jt j jt j jfkr |jt j jkr||s||Syt|}Wn6tk r}ztjd||j d|Wdd}~XYnX|||S|jt j jkr0tjd||j dn|jt j jkrN|||jS|j dkrhtjd|d|j|j |dd Stjk r}zs z$AstroidManager.file_from_module_namecCsp|p|j}||jkr|j|Sy |j}t|r<|||SWntk rRYnXddlm}|| ||S)z3given an imported module, return the astroid objectr)r) __name__r__file__rZis_python_sourcer)AttributeErrorr"rZ module_build)rrIr&r%rr r r rEs     zAstroidManager.ast_from_modulec Csh|dkrNy |j}Wn:tk rL}ztjd|t||d|Wdd}~XYnX||}||jdS)zget astroid for the given classNz,Unable to get module for class {class_name}.)cls class_reprr&r) __module__r]rr$r r#getattrr[)rklassr&exc modastroidr r r ast_from_classs  zAstroidManager.ast_from_classc cspt|drt|ts|j}n|}y |j}Wnntk rb}ztjd|t|d|Wdd}~XYn8t k r}ztj d|t|d|Wdd}~XYnXy |j }Wnptk r}ztjd|t|d|Wdd}~XYn:t k r}ztj d|t|d|Wdd}~XYnX| |}||krJx@| ||D] }|Vq8Wn"x | ||D]}|VqXWdS)z!infer astroid for the given class __class__z&Unable to get module for {class_repr}.)r^r_NzBUnexpected error while retrieving module for {class_repr}: {error}z%Unable to get name for {class_repr}: z@Unexpected error while retrieving name for {class_repr}: {error})hasattrrYr?rfr`r]rr$r rrDr[r#ZigetattrZinstantiate_class) rrcontextrbr&rcnamerdZinferredr r r infer_ast_from_somethingsJ        z'AstroidManager.infer_ast_from_somethingcCs|j|dS)a}Registers a hook to resolve imports that cannot be found otherwise. `hook` must be a function that accepts a single argument `modname` which contains the name of the module or package that could not be imported. If `hook` can resolve the import, must return a node of type `astroid.Module`, otherwise, it must raise `AstroidBuildingError`. N)rappend)rrLr r r register_failed_import_hook,sz*AstroidManager.register_failed_import_hookcCs|j|j|dS)zrErerjrlrnrprrr r r r r $s(   ?   ,  r ) rtr:rMrorZastroid.interpreter._importrrrr r r r r r s