a
XC?h @ sV d dl mZ ddlmZmZmZmZ d dlmZm Z m
Z
dd Z G dd dej Z
d S )
)
fixer_base )dirnamejoinexistssep)
FromImportsymstokenc c s | g}|r| }|jtjkr(|jV q|jtjkrNddd |jD V q|jtj krl|
|jd q|jtj kr| |jd d d qt
dqd S )N c S s g | ]
}|j qS )value).0Zchr r =/opt/alt/python39/lib64/python3.9/lib2to3/fixes/fix_import.py
z$traverse_imports..r zunknown node type)poptyper
NAMEr
r Z dotted_namer childrenZdotted_as_nameappendZdotted_as_namesextendAssertionError)namesZpendingnoder 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
Zchangedr Zwarningr prefix)r! r Zresultsr% Z
have_localZ
have_absoluteZmod_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_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)r r Zos.pathr r r r Z
fixer_utilr r r
r ZBaseFixr r r r r s