$34 GRAYBYTE WORDPRESS FILE MANAGER $73



���]�;�
@spdd
lm
Z

ddlmZ
ddlmZmZ
ddlmZ	
ddl
mZ
dd�Zd	d
�Z
dd�Zd
d�Zddd�
Zdd�Zdd�Zdd�Ze
�e�dd�Zdddd�Zdd�Zdd�Zdd d!�
Zd"d#�Zdd$d%�
Zdd&d'�
Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Z d2d3d4d5d6d7d8d9d:d;h
Z!d<d=�Z"d>a#d?a$d@a%dAa&dBdC�Z'dDdE�Z(dFdG�Z)dHdI�Z*dJdK�Z+dLdM�Z,dNdO�Z-dPdQ�Z.e	j/e	j0hZ1ddRdS�
Z2e	j0e	j/e	j3hZ4dTdU�Z5ddVdW�
Z6dS)X�)
�islice�
)
�token)�Leaf�Node)
�python_symbols)
�patcompcCs%tt
j|ttjd
�|
g�S)N�
=)r�syms�argumentrr�EQUAL)�keyword�value�r�//opt/alt/python35/lib64/python3.5/fixer_util.py�
KeywordArgs
	
rcCstt
jd
�S)N�
()rr�LPARrrrr�LParens
rcCstt
jd
�S)N�
))rr�RPARrrrr�RParens
rcCsft|t
�s|g
}t|
t
�s9d
|
_|
g
}
ttj|ttjddd
�
g
|
�S)N�
 r	�prefix)	�
isinstance�listrrr
�atomrrr)�target�sourcerrr�Assigns
	

	
		
rNcCstt
j|d
|
�
S)Nr)rr�NAME)�namerrrr�Name&sr"cCs|tt
jt�|
g�gS)
N)rr
�trailer�Dot)�obj�attrrrr�Attr*sr'cCstt
jd
�S)N�
,)rr�COMMArrrr�Comma.sr*cCstt
jd
�S)N�
.)rr�DOTrrrrr$2sr$cCsJtt
j|
j�|j�g�}|rF|jd
tt
j|��
|S)Nr)rr
r#�clone�insert_child�arglist)�argsZlparenZrparen�noderrr�ArgList6s$


r2cCs7tt
j|t|
�
g�}|dk	r3||_|S)
N)rr
�powerr2r)�	func_namer0rr1rrr�Call=s

	
r5cCstt
jd
�S)N�

)rr�NEWLINErrrr�NewlineDsr8cCstt
jd
�S)N�)rrr7rrrr�	BlankLineHsr:cCstt
j|d
|
�
S)Nr)rr�NUMBER)�
nrrrr�NumberLs
r=c

Cs1tt
jttjd
�|ttjd�g�S)N�
[�
])rr
r#rr�LBRACE�RBRACE)
Z
index_noderrr�	SubscriptOs

rBcCstt
j|d
|
�
S)Nr)rr�STRING)�stringrrrr�StringUsrEc	Cs

d
|_d|
_d|_t
tjd�}d|_t
tjd�}d|_||
||g}|r�d|_t
tjd�}d|_|jttj||g��

ttj|ttj	|�g�}ttj
t
tjd�|t
tjd�g�S)Nr9r�for�in�ifr>r?)
rrrr �appendrr
Zcomp_ifZ	listmakerZcomp_forrr@rA)	Zxp�fp�itZtestZfor_leafZin_leafZ
inner_argsZif_leaf�innerrrr�ListCompYs$	
	
	

	

	


	

	

$
	


rMcCsx|
D]}|j�
qWt
tjd
�t
tj|dd�
t
tjddd�
ttj|
�g}ttj|�}|S)N�fromrr�import)�removerrr rr
�import_as_names�import_from)Zpackage_nameZ
name_leafsZleaf�children�imprrr�
FromImportqs





rUc	Cs�|
d
j�}|j
tjkr1|j�}nttj|j�g
�}|
d}|rodd�|D�
}ttjtt|d�
t|d�
�ttj|
dj�||
dj�g�g
|�}|j	|_	|S)	Nr%�afterc
Ssg|]}
|
j��qSr)
r-)�.0r<rrr�
<listcomp>�s	z!ImportAndCall.<locals>.<listcomp>r
rZlparZrpar)
r-�typer
r/rr3r'r"r#r)r1�results�namesr%Z
newarglistrV�newrrr�
ImportAndCall�s






	Z

r]c

Cs�t|t
�r.|jt�t�gkr.d
St|t
�o�t|j�
dko�t|jdt�o�t|jdt
�o�t|jdt�o�|jdjdko�|jdjdkS)NT�r
r�rr)rrrSrr�lenrr)
r1rrr�is_tuple�s*







rac

Cszt|t
�oyt|j�
d
koyt|jdt�oyt|jdt�oy|jdjdkoy|jdjdkS)Nrr
r>r?���rb)rrr`rSrr)
r1rrr�is_list�s




rcc

Cstt
jt�|t�g�S)
N)rr
rrr)
r1rrr�parenthesize�s
rd�sortedr�set�any�all�tuple�sum�min�max�	enumerateccs4t||
�}x|r/|V
t||
�}qWdS)
N)
�getattr)r%r&�nextrrr�
attr_chain�s
	

rpzefor_stmt< 'for' any 'in' node=any ':' any* >
        | comp_for< 'for' any 'in' node=any any* >
     z�
power<
    ( 'iter' | 'list' | 'tuple' | 'sorted' | 'set' | 'sum' |
      'any' | 'all' | 'enumerate' | (any* trailer< '.' 'join' >) )
    trailer< '(' node=any ')' >
    any*
>
z`
power<
    ( 'sorted' | 'enumerate' )
    trailer< '(' arglist<node=any any*> ')' >
    any*
>
Fc
Cs�ts9t
jt�
at
jt�
at
jt�
ad
atttg}
xRt|
t|d��D]8\}}i}|j||�ra|d|krad
SqaWdS)NT�parentr1F)	�
pats_builtrZcompile_pattern�p0�p1�p2�ziprp�match)r1�patterns�patternrqrZrrr�in_special_context�s





%

"

rzc
Cs�|j}
|
dk	r+|
j
tjkr+d
S|j}|j
tjtjfkrSd
S|j
tjkr||j	d|kr|d
S|j
tj
ks�|j
tjkr�|
dk	r�|
j
tjks�|j	d|kr�d
SdS)NFr
T)
Zprev_siblingrYrr,rqr
�funcdef�classdef�	expr_stmtrSZ
parametersZ
typedargslistr))r1�prevrqrrr�is_probably_builtin�s	

	


%




rc
Csmxf|dk	rh|jt
jkr\t|j�
d
kr\|jd}
|
jtjkr\|
jS|j}qWdS)Nr_rr9)	rYr
�suiter`rSr�INDENTrrq)r1�indentrrr�find_indentation
s
'






r�c
CsW|jt
jkr|S|j�}|jd}
|_tt
j|g
�}|
|_|S)
N)rYr
r�r-rqr)r1rqr�rrr�
make_suite
s





	
r�c

Cs8x1|jt
jkr3|j}|std
�
�
qW|S)Nz,root found before file_input node was found.)rYr
Z
file_inputrq�
ValueError)
r1rrr�	find_root(
s

	


r�cCs"t|
t
|�
|�}t|�
S)
N)�find_bindingr��bool)�packager!r1Zbindingrrr�does_tree_import1
s
r�c

Cs|jt
jt
jfkS)
N)rYr
�import_namerR)
r1rrr�	is_import9
sr�cCs�
d
d�}t|�
}t
||
|�r.dSd}}xnt|j�
D]]\}}||�
scqHx4t|j|d��
D]\}}||�
s}Pq}W||}PqHW|dkr
x\t|j�
D]K\}}|jtjkr�|jr�|jdjtjkr�|d}Pq�W|dkrY
t	tj
ttjd�ttj|
dd�
g�}	n$t
|ttj|
dd�
g
�}	|	t�g}
|j|t	tj|
��
dS)Nc

Ss,|jt
jko+|jo+t|jd
�
S)Nr
)rYr
�simple_stmtrSr�)
r1rrr�is_import_stmt@
s

z$touch_import.<locals>.is_import_stmtr
rrOrr)r�r�rmrSrYr
r�rrCrr�rr rUr8r.)r�r!r1r��rootZ
insert_pos�offset�idxZnode2�import_rSrrr�touch_import=
s4





&











	

!$
r�cCs<x5|
jD]*}d}|j
tjkrpt||jd
�rB|St|t|jd�
|�}|r|}n�
|j
tjtjfkr�t|t|jd�
|�}|r|}n[
|j
tj	krv
t|t|jd�
|�}|r�|}qx
t
|jdd��
D]\\}}|j
tjkr
|j
dkr
t|t|j|d�
|�}|r
|}q
Wn�|j
tkr�
|jd
j
|kr�
|}npt|||�r�
|}nU|j
tjkr�
t|||�}n.|j
tjkrt||jd�r|}|r
|s$|St|�
r
|Sq
WdS)	Nrr_r^�
:�r
rbrb)rSrYr
Zfor_stmt�_findr�r�Zif_stmtZ
while_stmtZtry_stmtrmr�COLONr�	_def_syms�_is_import_bindingr�r}r�)r!r1r��child�retr<�
iZkidrrrr�k
sH





	


	



	&
!#


%
	

	









r�cCs||
g
}xl|rw|j�}
|
j
d
krO|
j
tkrO|j|
j�

q|
j
tjkr|
j|kr|
SqWdS)N�
)�poprY�_block_syms�extendrSrr r)r!r1Znodesrrrr��
s
	
	



!

r�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
j
kr�
t|
|�r�
|S|jt
jkr�
|jd}|jtjkr |j|
kr |SnA|jtjkr|j|
kr|S|r |jtjkr |SdS)Nrr_r^�asrb)rYr
r�rSZdotted_as_namesZdotted_as_namerrr rR�str�stripr�rQZimport_as_name�STAR)r1r!r�rTr�Zlastr<rrrr��
s@







!




!

!

%




!




!

!



r�)7�	itertoolsrZpgen2rZpytreerrZpygramrr
r9rrrrrr"r'r*r$r2r5r8r:r=rBrErMrUr]rarcrdZconsuming_callsrprsrtrurrrzrr�r�r�r�r�r�r|r{r�r�r#r�r�r�rrrr�<module>sZ





		-
*