a
i @ sZ d Z ddlmZ ddlmZmZmZmZ ddlm Z m
Z
m Z dd Z G dd d ej
Zd
S ) zFixer 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 )dirnamejoinexistssep)
FromImportsymstokenc c s | g}|r| }|jtjkr(|jV q|jtjkrNddd |jD V q|jtj krl|
|jd q|jtj kr| |jddd qt
dqdS )zF
Walks over all the names imported in a dotted_as_names node.
c S s g | ]
}|j qS )value).0Zchr r 0/usr/lib64/python3.9/lib2to3/fixes/fix_import.py
z$traverse_imports..r Nzunknown node type)poptyper
NAMEr
r Z dotted_namer childrenZdotted_as_nameappendZdotted_as_namesextendAssertionError)namespendingnoder r r traverse_imports s
r c s4 e Zd ZdZdZ fddZdd 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_featuresskip)selfZtreename __class__r r r / s zFixImport.start_treec C s | j r
d S |d }|jtjkrVt|ds4|jd }q | |jrd|j |_| nZd}d}t |D ]}| |rzd}qfd}qf|r|r|
|d d S t d|g}|j |_ |S d S )Nimpr
r .FTz#absolute and local imports together)
r! r r Z import_fromhasattrr probably_a_local_importr
changedr warningr prefix)r" r resultsr& Z
have_localZ
have_absolutemod_namenewr r r transform3 s,
zFixImport.transformc C st | drdS |ddd }t| j}t||}ttt|dsHdS dtddd d
fD ]}t|| rX d S qXdS ) Nr' F r z __init__.pyz.pyz.pycz.soz.slz.pydT)
startswithsplitr filenamer r r )r" Zimp_name base_pathextr r r r) U s
z!FixImport.probably_a_local_import) __name__
__module__ __qualname__Z
BM_compatibleZPATTERNr r0 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