Your IP : 216.73.216.196
B
��\ 1 � @ s^ d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ dd� Z G d d
� d
�Z
dS ) z�astroid 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)�
transformsc C s" yt | �S tk
r dS X d S )Nz???)�repr� Exception)�obj� r �@/opt/alt/python37/lib/python3.7/site-packages/astroid/manager.py� safe_repr s r c @ s� e Zd ZdZdZi Zdd� Zedd� �Zdd� Z d(d d
�Z
dd� Z dd� Z dd� Z
d)dd�Zdd� Zdd� Zd*dd�Zd+d d �Zd,d d �Zd 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 loaderc C s\ t j| _| jsXi | _i | _g | _d| _d| _t� | _ t
� � | _ | j j
| _
| j j| _d| _d S )NF�d )r �brain�__dict__�
astroid_cache�_mod_file_cache�_failed_import_hooks�always_load_extensionsZ optimize_ast�set�extension_package_whitelistr ZTransformVisitor�
_transformZregister_transformZunregister_transformZmax_inferable_values)�selfr r r
�__init__. s
zAstroidManager.__init__c C s
| j d S )N�builtins)r )r r r r
�builtins_module? s z AstroidManager.builtins_modulec C s | j �|�S )z8Visit the transforms and apply them to the given *node*.)r �visit)r �noder r r
�visit_transformsC s z AstroidManager.visit_transformsNTFc C s� yt j|dd�}d}W n t jk
r, Y nX |dkrdyd�t �|��}W n tk
rb |}Y nX || jkr�| j| j|kr�| j| S |r�ddlm } || ��
||�S |r�|r�| � |�S t j
d|d��dS ) z.given a module name, return the astroid objectT)Zinclude_no_extN�.r )�AstroidBuilderz"Unable to build an AST for {path}.)�path)r Zget_source_fileZ NoSourceFile�joinZmodpath_from_file� ImportErrorr �file�astroid.builderr Z
file_build�ast_from_module_namer �AstroidBuildingError)r �filepath�modname�fallback�sourcer r r r
�
ast_from_fileG s(
z AstroidManager.ast_from_filec C s ddl m} || ��d|�S )Nr )r � )r$ r � string_build)r r( r r r r
�_build_stub_moduleb s z!AstroidManager._build_stub_modulec C s ddl m} |||�S )Nr )� build_namespace_package_module)r$ r/ )r r( r r/ r r r
�_build_namespace_moduleg s z&AstroidManager._build_namespace_modulec sH �j r
dS t�|�rdS |�d�� t� �fdd�tdt� �d �D ��S )NTr c 3 s& | ] }d � � d|� ��jkV qdS )r N)r! r )�.0�x)�partsr r r
� <genexpr>s s z5AstroidManager._can_load_extension.<locals>.<genexpr>� )r r Zis_standard_module�split�any�range�len)r r( r )r3 r r
�_can_load_extensionl s
z"AstroidManager._can_load_extensionc C s� || j kr| j | S |dkr&| �|�S t�� }|rDt�tj�|�� �z��y.| �||�}|jt j
j kr�| � |j
�}|dk r~|S n�|jt j
jt j
jfk�r |jt j
jkr�| �|�s�| �|�S yt�|�}W n6 tk
r� } ztjd||j
d�|�W dd}~X Y nX | �||�S |jt j
jk�r0tjd||j
d��n |jt j
jk�rN| �||j�S |j
dk�rhtjd|d��| j|j
|dd �S tj k
�r� } z<x2| j D ](}y||�S tj k
�r� Y nX �q�W |�W dd}~X Y nX W dt�|� X dS )
z.given a module name, return the astroid object�__main__Nz&Loading {modname} failed with:
{error})r( r z)Unable to load compiled module {modname}.z'Can't find a file for module {modname}.)r( F)r) ) r r. �os�getcwd�chdirr �dirname�file_from_module_name�typer �
ModuleTypeZ PY_ZIPMODULE�zip_import_data�location� C_BUILTIN� C_EXTENSIONr: r Zload_module_from_namer r �AstroidImportError�ast_from_module� PY_COMPILEDZ PY_NAMESPACEr0 �submodule_search_locationsr+ r& r ) r r( � context_fileZold_cwdZ
found_spec�module�ex�e�hookr r r
r% w s`
z#AstroidManager.ast_from_module_namec
C s� t d kr d S ddlm} || �}x�dD ]�}y |�|tjj d�\}}W n tk
r\ w&Y nX yHt �|| �}|� tjjd�}|�
|�r�|d }|� |� |�||�} | S t
k
r� w&Y q&X q&W d S )Nr )r )z.zipz.eggr5 r z .__init__)� zipimportr$ r �rsplitr< r �sep�
ValueError� zipimporter�replace�
is_packager- �
get_sourcer )
r r' r Zbuilder�extZeggpath�resource�importerZzmodnamerL r r r
rC � s(
z AstroidManager.zip_import_datac
C s� y| j ||f }W nn tk
r� ytj|�d�|d�}W n2 tk
rl } ztjd||d�}W d d }~X Y nX || j ||f< Y nX t|tj �r�|�|S )Nr )rK z6Failed to import module {modname} with error:
{error}.)r( �error)
r �KeyErrorr Zfile_info_from_modpathr6 r" r rG �
isinstancer& )r r( Z contextfile�valuerM r r r
r@ � s z$AstroidManager.file_from_module_namec C sp |p|j }|| jkr | j| S y |j}t�|�r<| �||�S W n tk
rR Y nX ddlm} || �� ||�S )z3given an imported module, return the astroid objectr )r )
�__name__r �__file__r Zis_python_sourcer+ �AttributeErrorr$ r Z module_build)r rL r( r' r r r r
rH � s
z AstroidManager.ast_from_modulec
C sh |dkrNy
|j }W n: tk
rL } z tjd|t|�|d�|�W dd}~X Y nX | �|�}|�|j�d S )z get astroid for the given classNz,Unable to get module for class {class_name}.)�cls�
class_reprr( r )�
__module__ra r r& r r% �getattrr_ )r �klassr( �exc�
modastroidr r r
�ast_from_class� s
z AstroidManager.ast_from_classc
c sp t |d�r t|t�s |j}n|}y
|j}W nn tk
rb } ztjd|t|�d�|�W dd}~X Y n8 t k
r� } ztj
d|t|�d�|�W dd}~X Y nX y
|j }W np tk
r� } ztjd|t|�d�|�W dd}~X Y n: t k
�r } ztj
d|t|�d�|�W dd}~X Y nX | � |�}||k�rJx@|�
||�D ] }|V �q8W n"x |�
||�D ]}|�� V �qXW dS )z!infer astroid for the given class� __class__z&Unable to get module for {class_repr}.)rb rc 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})�hasattrr] rA rj rd ra r r& r r rG r_ r% ZigetattrZinstantiate_class) r r �contextrf r( rg �namerh �inferredr r r
�infer_ast_from_something sJ
z'AstroidManager.infer_ast_from_somethingc C s | 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)r �append)r rO r r r
�register_failed_import_hook, s z*AstroidManager.register_failed_import_hookc C s | j �|j|� dS )z<Cache a module if no module with the same name is known yet.N)r �
setdefaultrm )r rL r r r
� cache_module6 s zAstroidManager.cache_modulec C s ddl }|j�� dS )z�Bootstrap the required AST modules needed for the manager to work
The bootstrap usually involves building the AST for the builtins
module, which is required by the rest of astroid to work correctly.
r N)Zastroid.raw_buildingZ raw_buildingZ_astroid_bootstrapping)r �astroidr r r
� bootstrap: s zAstroidManager.bootstrapc C s | j �� | �� dS )z@Clear the underlying cache. Also bootstraps the builtins module.N)r �clearru )r r r r
� clear_cacheD s
zAstroidManager.clear_cache)NTF)N)N)N)N)r_ rd � __qualname__�__doc__rm r r �propertyr r r+ r. r0 r: r% rC r@ rH ri ro rq rs ru rw r r r r
r $ s(
?
,
r ) ry r< rP rt r Zastroid.interpreter._importr r r r r r r r r
�<module> s