��Qc@szdZd
dl
mZ
ddlmZmZmZmZ
d
dlm	Z	m
Z
mZ
d�Zdej
f
d��YZd	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�(tdirnametjointexiststsep(t
FromImporttsymsttokenc
cs�|g
}
x�|
r�
j�}|j
tjkr;|jV
q|j
tjkrwd
jg|jD]}|j^q]�V
q|j
tj	kr�|
j
|jd�
q|j
tjkr�|
j|jddd��
qt
d��qWdS(sF
    Walks over all the names imported in a dotted_as_names node.
    tiNi�sunkown node type(tpopttypeRtNAMEtvalueRtdotted_nameRtchildrentdotted_as_nametappendtdotted_as_namestextendtAssertionError(tnamestpendingtnodetch((s./sys/lib/python2.7/lib2to3/fixes/fix_import.pyttraverse_importss	
	




*



 t	FixImportc
Bs/eZ
eZdZd
�Zd�Zd�ZRS(sj
    import_from< 'from' imp=any 'import' ['('] any [')'] >
    |
    import_name< 'import' imp=any >
    cCs/tt
|�j|
|�
d
|
jk|_dS(Ntabsolute_import(tsuperRt
start_treetfuture_featurestskip(tselfttreetname((s./sys/lib/python2.7/lib2to3/fixes/fix_import.pyR/s

cCs
|jr
dS|d
}|
j
tjkr~x t|d�sK|jd}q,W|j|j�r
d|j|_|j�
q
n�t	}t	}x2t
|�D]$}|j|�r�t}q�t}q�W|r�r�|j|
d�
ndSt
d|g
�}|
j|_|SdS(NtimpRiu
.s#absolute and local imports together(RR
Rtimport_fromthasattrRtprobably_a_local_importRtchangedtFalseRtTruetwarningRtprefix(RRtresultsR"t
have_localt
have_absolutetmod_nametnew((s./sys/lib/python2.7/lib2to3/fixes/fix_import.pyt	transform3s,
	










	





cCs�|
jd
�rt
S|
jd
d�d}
t|j�}t||
�}ttt|�d��sft
Sx4dtdddd	gD]}t||�rtSqWt
S(
Nu
.i
is__init__.pys.pys.pycs.sos.sls.pyd(	t
startswithR'tsplitR
tfilenameRRRR((Rtimp_namet	base_pathtext((s./sys/lib/python2.7/lib2to3/fixes/fix_import.pyR%Us








(t__name__t
__module__R(t
BM_compatibletPATTERNRR0R%(((s./sys/lib/python2.7/lib2to3/fixes/fix_import.pyR&s

		"N(t__doc__RRtos.pathR
RRRt
fixer_utilRRRRtBaseFixR(((s./sys/lib/python2.7/lib2to3/fixes/fix_import.pyt<module>s

"