File manager

U

e5d��@sVdd
lm
Z

ddlmZmZmZmZ
ddlmZm	Z	m
Z

dd�ZGdd�de
j�Z
d	S)
�)
�
fixer_base�)�dirname�join�exists�sep)�
FromImport�syms�tokenc
cs�|g
}
|
r�|
��}|j
tjkr(|jV
q|j
tjkrNd
�dd�|jD�
�
V
q|j
tj	krl|
�
|jd�

q|j
tjkr�|
�|jddd��

qt
d�
�
qdS)N�c
Ssg|]
}
|
j�qS�)
�value)�.0Zchrr�0/usr/lib64/python3.8/lib2to3/fixes/fix_import.py�
<listcomp>sz$traverse_imports.<locals>.<listcomp>r���zunknown node type)�pop�typer
�NAMEr
r	Zdotted_namer�childrenZdotted_as_name�appendZdotted_as_names�extend�AssertionError)�namesZpending�noderrr�traverse_importss










rcs4eZ
dZd
ZdZ�f
dd�Zdd�Zdd�Z�ZS)	�	FixImportTzj
    import_from< 'from' imp=any 'import' ['('] any [')'] >
    |
    import_name< 'import' imp=any >
    cs"tt
|��|
|�
d
|
jk|_dS)NZabsolute_import)�superr�
start_treeZfuture_features�skip)�selfZtree�name�
�	__class__rrr/s

zFixImport.start_treecCs�|jr
dS|d
}|
j
tjkrVt|d�s4|jd}q|�|j�
r�d|j|_|��
nZd}d}t	|�
D]}|�|�
rzd}qfd}qf|r�|r�|�
|
d�
dStd|g
�}|
j|_|SdS)N�impr
r�
.FTz#absolute and local imports together)
rrr	Zimport_from�hasattrr�probably_a_local_importr
ZchangedrZwarningr�prefix)r rZresultsr$Z
have_localZ
have_absoluteZmod_name�newrrr�	transform3s,


















zFixImport.transformcCst|
�d
�
rdS|
�
d
d�d}
t|j�
}t||
�}ttt|�
d��
sHdSdtddd	d
fD]}t||�
rX
dSqXdS)Nr%F�
rz__init__.pyz.pyz.pycz.soz.slz.pydT)�
startswith�splitr�filenamerrr)r Zimp_name�	base_pathZextrrrr'Us











z!FixImport.probably_a_local_import)	�__name__�
__module__�__qualname__Z
BM_compatibleZPATTERNrr*r'�
__classcell__rrr"rr&s

"rN)rrZos.pathrrrrZ
fixer_utilrr	r
rZBaseFixrrrrr�<module>s