a
XC?h� � @ 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}|r�|� � }|jtjkr(|jV q|jtjkrNd�dd� |jD ��V q|jtj krl|�
|jd � q|jtj kr�|� |jddd� � qt
d��qdS )zF
Walks over all the names imported in a dotted_as_names node.
� c S s g | ]
}|j �qS � )�value)�.0Zchr r �=/opt/alt/python39/lib64/python3.9/lib2to3/fixes/fix_import.py�
� 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|jv | _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 }|jtjkrVt|d�s4|jd }q | �|j�r�d|j |_|�� nZd}d}t |�D ]}| �|�rzd}qfd}qf|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 st |� d�rdS |�dd�d }t| j�}t||�}ttt|�d��sHdS dtddd d
fD ]}t|| �rX d S qXdS ) 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