�
{ fc @ sz d Z d d l m Z d d l m Z m Z m Z m Z d d l m Z m
Z
m Z d � Z d e j
f d � � YZ d S(
s� 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
i ( t
fixer_basei����( t dirnamet joint existst sep( t
FromImportt symst tokenc c s� | g } x� | r� | j � } | j t j k r; | j Vq | j t j k rw d j g | j D] } | j ^ q] � Vq | j t j k r� | j
| j d � q | j t j k r� | j | j d d d � � q t
d � � q Wd S( sF
Walks over all the names imported in a dotted_as_names node.
t i Ni����s unknown node type( t popt typeR t NAMEt valueR t dotted_nameR t childrent dotted_as_namet appendt dotted_as_namest extendt AssertionError( t namest pendingt nodet ch( ( s0 /usr/lib64/python2.7/lib2to3/fixes/fix_import.pyt traverse_imports s * t FixImportc B s/ e Z e Z d Z d � Z d � Z d � Z RS( sj
import_from< 'from' imp=any 'import' ['('] any [')'] >
|
import_name< 'import' imp=any >
c C s/ t t | � j | | � d | j k | _ d S( Nt absolute_import( t superR t
start_treet future_featurest skip( t selft treet name( ( s0 /usr/lib64/python2.7/lib2to3/fixes/fix_import.pyR / s c C s | j r
d S| d } | j t j k r~ x t | d � sK | j d } q, W| j | j � rd | j | _ | j � qn� t } t } x2 t
| � D]$ } | j | � r� t } q� t } q� W| r� | r� | j | d � n d St
d | g � } | j | _ | Sd S( Nt impR i u .s# absolute and local imports together( R R
R t import_fromt hasattrR t probably_a_local_importR t changedt FalseR t Truet warningR t prefix( R R t resultsR" t
have_localt
have_absolutet mod_namet new( ( s0 /usr/lib64/python2.7/lib2to3/fixes/fix_import.pyt transform3 s,
c C s� | j d � r t S| j d d � d } t | j � } t | | � } t t t | � d � � sf t Sx4 d t d d d d g D] } t | | � r t Sq Wt S(
Nu .i i s __init__.pys .pys .pycs .sos .sls .pyd( t
startswithR' t splitR t filenameR R R R( ( R t imp_namet base_patht ext( ( s0 /usr/lib64/python2.7/lib2to3/fixes/fix_import.pyR% U s ( t __name__t
__module__R( t
BM_compatiblet PATTERNR R0 R% ( ( ( s0 /usr/lib64/python2.7/lib2to3/fixes/fix_import.pyR & s
"N( t __doc__R R t os.pathR R R R t
fixer_utilR R R R t BaseFixR ( ( ( s0 /usr/lib64/python2.7/lib2to3/fixes/fix_import.pyt s
"