B a8@sdZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZdZd d ZGd d d ZdS) 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)ClassVar)AstroidBuildingErrorAstroidImportError)spec) NoSourceFilefile_info_from_modpathget_source_fileis_python_sourceis_standard_moduleload_module_from_namemodpath_from_file)TransformVisitor)z.zipz.eggz.whlcCs"yt|Stk rdSXdS)Nz???)repr Exception)objr3/tmp/pip-unpacked-wheel-mm06h1t3/astroid/manager.py safe_repr2src@seZdZUdZdZiZdZeee d<ddZ e ddZ e d d Z e d d Zd dZd1ddZd2ddZddZddZddZd3ddZdd Zd!d"Zd4d#d$Zd5d%d&Zd6d'd(Zd)d*Zd+d,Zd-d.Zd/d0ZdS)7AstroidManagerz_Responsible to build astroid from files or modules. Use the Borg (singleton) pattern. zastroid loaderdmax_inferable_valuescCs@tj|_|jssz5AstroidManager._can_load_extension..)rr splitanyrangelen)r r2r)r?r r_can_load_extensions  z"AstroidManager._can_load_extensionc Cs||jkr|j|S|dkr&||S|rDt}ttj|zy@|||}|jt j j kr| |j }|dk r~|Sn|jt j jt j jfkr|jt j jkr||s||Sy t|}Wn4tk r}ztd||j d|Wdd}~XYnX|||S|jt j jkr*td||j dn8|jt j jkrH|||jS|jt j jkrb||S|j dkrztd|d|j|j |dd Stk r}z:x0|jD]&}y||Stk rYnXqW|Wdd}~XYnXWd|rt|XdS) z.given a module name, return the astroid object__main__Nz&Loading {modname} failed with: {error})r2r+z)Unable to load compiled module {modname}.z'Can't find a file for module {modname}.)r2F)r3)rr:osgetcwdchdirr+dirnamefile_from_module_nametyper ModuleTypeZ PY_ZIPMODULEzip_import_datalocationZ C_BUILTIN C_EXTENSIONrFr rrast_from_module PY_COMPILEDZ PY_NAMESPACEr<submodule_search_locationsZ PY_FROZENr5rr)r r2 context_fileZold_cwdZ found_specmoduleehookrrrr0sf                z#AstroidManager.ast_from_module_namec Cstdkr dSddlm}||}xtD]}y||tjjd\}}Wntk r\w&YnXyHt ||}| tjjd}| |r|d}| | |||} | Stk rw&Yq&Xq&WdS)Nr)r*rAr)z .__init__) zipimportr/r*ZIP_IMPORT_EXTSrsplitrHr+sep ValueError zipimporterreplace is_packager7 get_sourcer) r r1r*builderextZeggpathresourceimporterZzmodnamerVrrrrOs(    zAstroidManager.zip_import_datac Csy|j||f}Wnptk ryt|d|d}Wn6tk rn}ztd||dd}Wdd}~XYnX||j||f<YnXt|tr|d|S)Nr))rUz6Failed to import module {modname} with error: {error}.)r2error) rKeyErrorrrBr-rwith_traceback isinstancer)r r2Z contextfilevaluerWrrrrLs   z$AstroidManager.file_from_module_namecCsn|p|j}||jkr|j|Sy|j}t|r:|||SWntk rPYnXddlm}||||S)z3given an imported module, return the astroid objectr)r*) __name__r__file__r r5AttributeErrorr/r*Z module_build)r rVr2r1r*rrrrR s    zAstroidManager.ast_from_modulec Csf|dkrLy |j}Wn8tk rJ}ztd|t||d|Wdd}~XYnX||}||jdS)zget astroid for the given classNz,Unable to get module for class {class_name}.)cls class_reprr2r) __module__rmrrr0getattrrk)r klassr2exc modastroidrrrast_from_classs  zAstroidManager.ast_from_classc csht|drt|ts|j}n|}y |j}Wnjtk r`}ztd|t|d|Wdd}~XYn6tk r}zt d|t|d|Wdd}~XYnXy |j }Wnltk r}ztd|t|d|Wdd}~XYn8tk r }zt d|t|d|Wdd}~XYnX| |}||krBx@| ||D] }|Vq0Wn"x | ||D]}| VqPWdS)z!infer astroid for the given class __class__z&Unable to get module for {class_repr}.)rnroNzBUnexpected 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})hasattrrirMrvrprmrrrrrkr0ZigetattrZinstantiate_class) r rcontextrrr2rsnamertinferredrrrinfer_ast_from_something,sJ        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)r rXrrrregister_failed_import_hookXsz*AstroidManager.register_failed_import_hookcCs|j|j|dS)zs  $