B
� f� � @ sZ d Z ddlmZ ddlmZmZmZmZ ddlm Z m
Z
m Z dd� Z G dd � d ej
�Zd
S ) z�Fixer for import statements.
If spam is being imported from the local directory, this import:
from spam import eggs
Becomes:
from .spam import eggs
And this import:
import spam
Becomes:
from . import spam
� )�
fixer_base� )�dirname�join�exists�sep)�
FromImport�syms�tokenc c s� | g}x�|r�|� � }|jtjkr*|jV q|jtjkrPd�dd� |jD ��V q|jtj krn|�
|jd � q|jtj kr�|� |jddd� � qt
d��qW dS )zF
Walks over all the names imported in a dotted_as_names node.
� c S s g | ]
}|j �qS � )�value)�.0Zchr r �=/opt/alt/python37/lib64/python3.7/lib2to3/fixes/fix_import.py�
s z$traverse_imports..r N�����zunknown node type)�pop�typer
�NAMEr
r Z dotted_namer �childrenZdotted_as_name�appendZdotted_as_names�extend�AssertionError)�namesZpending�noder r r �traverse_imports s
r c s4 e Zd ZdZdZ� fdd�Zdd� Zdd� Z� ZS ) � FixImportTzj
import_from< 'from' imp=any 'import' ['('] any [')'] >
|
import_name< 'import' imp=any >
c s" t t| ��||� d|jk| _d S )NZabsolute_import)�superr �
start_treeZfuture_features�skip)�selfZtree�name)� __class__r r r / s zFixImport.start_treec C s� | j r
d S |d }|jtjkrZxt|d�s6|jd }q W | �|j�r�d|j |_|�� n^d}d}x$t |�D ]}| �|�r�d}qld}qlW |r�|r�| �
|d� d S t d|g�}|j |_ |S d S )N�impr
r �.FTz#absolute and local imports together)
r r r Z import_from�hasattrr �probably_a_local_importr
Zchangedr Zwarningr �prefix)r r Zresultsr# Z
have_localZ
have_absoluteZmod_name�newr r r � transform3 s,
zFixImport.transformc C sv |� d�rdS |�dd�d }t| j�}t||�}ttt|�d��sHdS x(dtddd d
gD ]}t|| �rZd S qZW dS ) Nr$ F� r z __init__.pyz.pyz.pycz.soz.slz.pydT)�
startswith�splitr �filenamer r r )r Zimp_name� base_pathZextr r r r& U s
z!FixImport.probably_a_local_import) �__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr r) r& �
__classcell__r r )r" r r & s
"r N)�__doc__r r Zos.pathr r r r Z
fixer_utilr r r
r ZBaseFixr r r r r � s