��Qc@sdZd
dl
mZ
ddlmZ
ddlmZmZ
ddlm	Z

ddlmZ
d�Z
d	�Zd
�Zd�Zd5d�Zd
�Zd�Zd�Ze�e�d�Zd5d5d�Zd�Zd�Zd5d�Zd�Zd5d�Zd5d�Zd�Zd�Zd�Z d�Z!e"ddddd d!d"d#d$d%g
�Z#d&�Z$d'a%d(a&d)a'e(a)d*�Z*d+�Z+d,�Z,d-�Z-d.�Z.d/�Z/d0�Z0d1�Z1e"e
j2e
j3g�Z4d5d2�Z5e"e
j3e
j2e
j6g�Z7d3�Z8d5d4�Z9d5S(6s1Utility functions, node construction macros, etc.i�(
tislicei
(
ttoken(tLeaftNode(
tpython_symbols(
tpatcompcCs%tt
j|ttjd
�|
g�S(Nu
=(RtsymstargumentRR
tEQUAL(tkeywordtvalue((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt
KeywordArgs
	
cCstt
jd
�S(Nu
((RR
tLPAR(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytLParens
cCstt
jd
�S(Nu
)(RR
tRPAR(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytRParens
cCslt|t
�s|g
}nt|
t
�s?d
|
_|
g
}
nttj|ttjddd
�
g
|
�S(sBuild an assignment statementu
 u
=tprefix(	t
isinstancetlistRRRtatomRR
R(ttargettsource((s(/sys/lib/python2.7/lib2to3/fixer_util.pytAssigns


	
	
cCstt
j|d
|
�
S(sReturn a NAME leafR(RR
tNAME(tnameR((s(/sys/lib/python2.7/lib2to3/fixer_util.pytName&scCs|tt
jt�|
g�gS(
sA node tuple for obj.attr(RRttrailertDot(tobjtattr((s(/sys/lib/python2.7/lib2to3/fixer_util.pytAttr*scCstt
jd
�S(sA comma leafu
,(RR
tCOMMA(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytComma.scCstt
jd
�S(sA period (.) leafu
.(RR
tDOT(((s(/sys/lib/python2.7/lib2to3/fixer_util.pyR2scCsMtt
j|
j�|j�g�}|rI|jd
tt
j|��
n|S(s-A parenthesised argument list, used by Call()i
(RRRtclonetinsert_childtarglist(targstlparentrparentnode((s(/sys/lib/python2.7/lib2to3/fixer_util.pytArgList6s$


cCs:tt
j|t|
�g�}|d
k	r6||_n|S(sA function callN(RRtpowerR)tNoneR(t	func_nameR%RR(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytCall=s


cCstt
jd
�S(sA newline literalu

(RR
tNEWLINE(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytNewlineDscCstt
jd
�S(sA blank lineu(RR
R.(((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt	BlankLineHscCstt
j|d
|
�
S(NR(RR
tNUMBER(t
nR((s(/sys/lib/python2.7/lib2to3/fixer_util.pytNumberLs
c

Cs1tt
jttjd
�|ttjd�g�S(sA numeric or string subscriptu
[u
](RRRRR
tLBRACEtRBRACE(
t
index_node((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt	SubscriptOs

cCstt
j|d
|
�
S(s
A string leafR(RR
tSTRING(tstringR((s(/sys/lib/python2.7/lib2to3/fixer_util.pytStringUsc	Cs
d
|_d|
_d|_t
tjd�}d|_t
tjd�}d|_||
||g}|r�d|_t
tjd�}d|_|jttj||g��
nttj|ttj	|�g�}ttj
t
tjd�|t
tjd�g�S(suA list comprehension of the form [xp for fp in it if test].

    If test is None, the "if test" part is omitted.
    uu
 uforuinuifu
[u
](
RRR
RtappendRRtcomp_ift	listmakertcomp_forRR4R5(	txptfptitttesttfor_leaftin_leaft
inner_argstif_leaftinner((s(/sys/lib/python2.7/lib2to3/fixer_util.pytListCompYs$	
	
	

	

	


	

	
"
$
	


cCsx|
D]}|j�
qWt
tjd
�t
tj|dd�
t
tjddd�
ttj|
�g}ttj|�}|S(sO Return an import statement in the form:
        from package import name_leafsufromRu
 uimport(tremoveRR
RRRtimport_as_namestimport_from(tpackage_namet
name_leafstleaftchildrentimp((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt
FromImportqs





c

	Cs�t|t
�r.|jt�t�gkr.tSt|t
�o�t|j�d
ko�t|jdt�o�t|jdt
�o�t|jdt�o�|jdjdko�|jdjdkS(s(Does the node represent a tuple literal?iii
iu
(u
)(	RRROR
RtTruetlenRR
(
R(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytis_tuple�s*







c

Cszt|t
�oyt|j�d
koyt|jdt�oyt|jdt�oy|jdjdkoy|jdjdkS(s'Does the node represent a list literal?i
ii�u
[u
](RRRSRORR
(
R(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytis_list�s




c

Cstt
jt�|t�g�S(
N(RRRR
R(
R(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytparenthesize�s
tsortedRtsettanytallttupletsumtmintmaxt	enumerateccs4t||
�}x|r/|V
t||
�}qWd
S(sl
Follow an attribute chain.

    If you have a chain of objects where a.foo -> b, b.foo-> c, etc,
    use this to iterate over all objects in the chain. Iteration is
    terminated by getattr(x, attr) is None.

    Args:
        obj: the starting object
        attr: the name of the chaining attribute

    Yields:
        Each successive object in the chain.
    N(
tgetattr(RRtnext((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt
attr_chain�s
	

sefor_stmt< 'for' any 'in' node=any ':' any* >
        | comp_for< 'for' any 'in' node=any any* >
     s�
power<
    ( 'iter' | 'list' | 'tuple' | 'sorted' | 'set' | 'sum' |
      'any' | 'all' | 'enumerate' | (any* trailer< '.' 'join' >) )
    trailer< '(' node=any ')' >
    any*
>
s`
power<
    ( 'sorted' | 'enumerate' )
    trailer< '(' arglist<node=any any*> ')' >
    any*
>
c
Cs�ts<t
jt�at
jt�at
jt�atantttg}
xRt|
t|d
��D]8\}}i}|j	||�rd|d|krdtSqdWt
S(s
 Returns true if node is in an environment where all that is required
        of it is being iterable (ie, it doesn't matter if it returns a list
        or an iterator).
        See test_map_nochange in test_fixers.py for some examples and tests.
        tparentR((t
pats_builtRtcompile_patterntp0tp1tp2RRtzipRbtmatchtFalse(R(tpatternstpatternRctresults((s(/sys/lib/python2.7/lib2to3/fixer_util.pytin_special_context�s



	

%

"

c
Cs�|j}
|
dk	r+|
jtjkr+tS|j}|jtjtj	fkrStS|jtj
kr||jd
|kr|tS|jtjks�|jtj
kr�|
dk	r�|
jtjks�|jd
|kr�tStS(sG
    Check that something isn't an attribute or function name etc.
    iN(tprev_siblingR+ttypeR
R!RkRcRtfuncdeftclassdeft	expr_stmtROt
parameterst
typedargslistRRR(R(tprevRc((s(/sys/lib/python2.7/lib2to3/fixer_util.pytis_probably_builtin�s	

	


%




c
Cspxi|dk	rk|j
tjkr_t|j�d
kr_|jd}
|
j
tjkr_|
jSn|j	}qWdS(sFind the indentation of *node*.ii
uN(
R+RqRtsuiteRSROR
tINDENTR
Rc(R(tindent((s(/sys/lib/python2.7/lib2to3/fixer_util.pytfind_indentation�s
'







c
CsW|jt
jkr|S|j�}|jd}
|_tt
j|g
�}|
|_|S(
N(RqRRyR"RcR+R(R(RcRy((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt
make_suite
s





	
c

Cs;x4|jt
jkr6|j}|std
��qqW|S(sFind the top level namespace.s,root found before file_input node was found.(RqRt
file_inputRct
ValueError(
R(((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt	find_root
s

	


cCs"t|
t
|�|�}t|�S(
s� Returns true if name is imported from package at the
        top level of the tree which node belongs to.
        To cover the case of an import like 'import foo', use
        None for the package and 'foo' for the name. (tfind_bindingR�tbool(tpackageRR(tbinding((s(/sys/lib/python2.7/lib2to3/fixer_util.pytdoes_tree_import
s
c

Cs|jt
jt
jfkS(
s0Returns true if the node is an import statement.(RqRtimport_nameRK(
R(((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt	is_import"
scCs�d
�}t|�}t
||
|�r+dSd}}xnt|j�D]]\}}||�scqEnx1t|j|�D]\}}||�swPqwqwW||}PqEW|dkr
xbt|j�D]N\}}|jtjkr�|jr�|jdjtjkr�|d}Pq�q�Wn|dkr\
t
tjttj
d�ttj
|
dd�
g�}	n$t|ttj
|
dd�
g
�}	|	t�g}
|j|t
tj|
��
dS(s\ Works like `does_tree_import` but adds an import statement
        if it was not imported. c

Ss,|jt
jko+|jo+t|jd
�S(Ni(RqRtsimple_stmtROR�(
R(((s(/sys/lib/python2.7/lib2to3/fixer_util.pytis_import_stmt)
s

Nii
uimportRu
 (R�R�R_RORqRR�R
R8R+RR�RRRQR/R#(R�RR(R�troott
insert_postoffsettidxtnode2timport_RO((s(/sys/lib/python2.7/lib2to3/fixer_util.pyttouch_import&
s4	





 











	

!$
c	CsKxD|
jD]9}d}|jtjkrst||jd
�rB|St|t|jd�|�}|r |}q n�|jtjtj	fkr�t|t|jd�|�}|r |}q na
|jtj
kr|
t|t|jd�|�}|r�}q x
t|jd�D]b\}}|jtj
kr
|jdkr
t|t|j|d�|�}|ru
|}qu
q
q
Wn�|jtkr�|jd
j|kr�|}nvt|||�r�|}n[|jtjkr�|||�}n4|jtjkr t||jd�r |}q n|r
|s0|St|�rC|Sq
q
WdS(	s� Returns the node which binds variable name, otherwise None.
        If optional argument package is supplied, only imports will
        be returned.
        See test cases for examples.i
i�iit
:iiN(ROR+RqRtfor_stmtt_findR�R}tif_stmtt
while_stmtttry_stmtR_R
tCOLONR
t	_def_symst_is_import_bindingR�RtR�(RR(R�tchildtretR2t
itkid((s(/sys/lib/python2.7/lib2to3/fixer_util.pyR�T
sH












	 
!#

%
	

	









cCs||
g
}xl|rw|j�}
|
j
d
krO|
j
tkrO|j|
j�
q|
j
tjkr|
j|kr|
SqWdS(Ni
(	tpopRqt_block_symstextendROR
RR
R+(RR(tnodes((s(/sys/lib/python2.7/lib2to3/fixer_util.pyR�
s
	
	



!

cCs'|jt
jkr
|r
|jd
}|jt
jkr�x�|jD]Z}|jt
jkrw|jdj|
kr�|SqB|jtjkrB|j|
krB|SqBWq#|jt
jkr�jd}|jtjkr
|j|
kr
|Sq#|jtjkr#|j|
kr#|Sn
|jt
j	kr#|rM
t
|jd
�j�|krM
dS|jd}|rs
t
d|�rs
dS|jt
jkr�t
|
|�r�|S|jt
jkr�|jd}|jtjkr |j|
kr |Sq#|jtjkr|j|
kr|S|r#|jtjkr#|SndS(s� Will reuturn node if node will import name, or node
        will import * from package.  None is returned otherwise.
        See test cases for examples. i
ii�iuasN(RqRR�ROtdotted_as_namestdotted_as_nameR
R
RRKtunicodetstripR+R�RJtimport_as_nametSTAR(R(RR�RPR�tlastR2((s(/sys/lib/python2.7/lib2to3/fixer_util.pyR��s@







!




!

!

%




!




!

!



N(:t__doc__t	itertoolsRtpgen2R
tpytreeRRtpygramRRtRRR
RRR+RRR RR)R-R/R0R3R7R:RHRQRTRURVRXtconsuming_callsRbRfRgRhRkRdRoRxR|R}R�R�R�R�RsRrR�R�RR�R�R�(((s(/sys/lib/python2.7/lib2to3/fixer_util.pyt<module>
sZ


														
	

										-
*