File: //usr/local/lib/python3.10/lib2to3/fixes/__pycache__/fix_import.cpython-310.opt-2.pyc
o
�i� � @ sX d dl mZ ddlmZmZmZmZ d dlmZm Z m
Z
dd� Z G dd� dej �Z
d S )
� )�
fixer_base� )�dirname�join�exists�sep)�
FromImport�syms�tokenc c s� � | g}|rR|� � }|jtjkr|jV n8|jtjkr)d�dd� |jD ��V n%|jtj kr8|�
|jd � n|jtj krJ|� |jd d d� � nt
d��|sd S d S )N� c S s g | ]}|j �qS � )�value)�.0Zchr r �5/usr/local/lib/python3.10/lib2to3/fixes/fix_import.py�
<listcomp> s z$traverse_imports.<locals>.<listcomp>r ���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 )NZabsolute_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_absoluteZmod_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 r. r( �
__classcell__r r r# r r & s "r N)r r Zos.pathr r r r Z
fixer_utilr r r
r ZBaseFixr r r r r �<module> s