File: //opt/saltstack/salt/lib/python3.10/lib2to3/fixes/__pycache__/fix_import.cpython-310.pyc
o
;j� � @ 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}|rQ|� � }|jtjkr|jV n8|jtjkr(d�dd� |jD ��V n%|jtj kr7|�
|jd � n|jtj krI|� |jddd� � nt
d��|sdS dS )zF
Walks over all the names imported in a dotted_as_names node.
� c S s g | ]}|j �qS � )�value)�.0�chr r �>/opt/saltstack/salt/lib/python3.10/lib2to3/fixes/fix_import.py�
<listcomp> s z$traverse_imports.<locals>.<listcomp>r N���zunknown node type)�pop�typer
�NAMEr
r Z dotted_namer �childrenZdotted_as_name�appendZdotted_as_names�extend�AssertionError)�names�pending�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 )N�absolute_import)�superr �
start_treeZfuture_features�skip)�selfZtree�name�� __class__r r r! / s zFixImport.start_treec C s� | j rd S |d }|jtjkr2t|d�s |jd }t|d�r| �|j�r0d|j |_|�� d S d S d}d}t |�D ] }| �|�rDd}q:d}q:|rS|rQ| �
|d� d S t d|g�}|j |_ |S )N�impr
r �.FTz#absolute and local imports together)
r" r r Z import_from�hasattrr �probably_a_local_importr
�changedr �warningr �prefix)r# r �resultsr' Z
have_localZ
have_absolute�mod_name�newr r r � transform3 s0
� �
zFixImport.transformc C st |� d�rdS |�dd�d }t| j�}t||�}ttt|�d��s$dS dtddd d
fD ] }t|| �r7 d S q,dS ) Nr( F� r z __init__.pyz.pyz.pycz.soz.slz.pydT)�
startswith�splitr �filenamer r r )r# Zimp_name� base_path�extr r r r* U s
�z!FixImport.probably_a_local_import) �__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr! r1 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 �<module> s