$21 GRAYBYTE WORDPRESS FILE MANAGER $83

�
j f��@s�dZd
dl
mZ
d
dlmZ
d
dl
mZ
d
dlmZmZm	Z	m
Z
mZmZ
dd�Z
Gdd	�d	ej�Zd
d�Zdd
�Zgddd�Zdd�ZdS)a:
Fixer for function definitions with tuple parameters.

def func(((a, b), c), d):
    ...

    ->

def func(x, d):
    ((a, b), c) = x
    ...

It will also support lambdas:

    lambda (x, y): x + y -> lambda t: t[0] + t[1]

    # The parens are a syntax error in Python 3
    lambda (x): x + y -> lambda x: x + y
�)
�pytree)
�token)
�
fixer_base)�Assign�Name�Newline�Number�	Subscript�symsc

Cs)t|t
j�o(|jd
jtjkS)N�)�
isinstancer�Node�children�typer�STRING)
�stmt�r�C/opt/alt/python34/lib64/python3.4/lib2to3/fixes/fix_tuple_params.py�is_docstrings

rc@s:eZ
dZd
ZdZdZdd�Zdd�ZdS)	�FixTupleParams�Ta

              funcdef< 'def' any parameters< '(' args=any ')' >
                       ['->' any] ':' suite=any+ >
              |
              lambda=
              lambdef< 'lambda' args=vfpdef< '(' inner=any ')' >
                       ':' body=any
              >
              cs6d
|kr�j|
|�Sg�
|d}|d}|dj
djtjkryd}|dj
dj}t��n!d}d}tjtjd��d	��
�fd
d�
}|jt	j
kr�||�

n`|jt	jkr4
xKt|j
�
D]7\}}	|	jt	j
kr�||	d|dk�


q�q�Wn�
s>
dSx�
D]}
|d|
_
qE
W|}|dkr~
d
�
d_n1t|dj
|�
r�
|�
d_|d}nx�
D]}
|d|
_
q�
W�
|dj
||�<x=t|d|t�
�
d�D]}||dj
|_qW|dj�
dS)N�lambda�suite�argsr�
r
z; �Fcs�t�j
��
}|j�}d
|_t||j��}|
rNd|_n|j|�

�
jtjt	j
|�j�g��

dS)Nr�
 )r�new_name�clone�prefixr�replace�appendrr
r
Zsimple_stmt)Z	tuple_arg�
add_prefix�
n�argr)�end�	new_lines�selfrr�handle_tupleCs


	






z.FixTupleParams.transform.<locals>.handle_tupler"r)�transform_lambdarrr�INDENT�valuerrZLeafr
ZtfpdefZ
typedargslist�	enumerate�parentrr�range�lenZchanged)r'�node�resultsrr�start�indentr(�
ir$�line�afterr)r%r&r'r�	transform.sF















 













(

zFixTupleParams.transformc
CsB
|d
}|d}t|d�
}|j
tjkr\|j�}d|_|j|�

dSt|�
}t|�
}|j	t
|�
�
}t|dd�

}	|j|	j��

x�|j�D]}
|
j
tjkr�|
j
|kr�dd�||
j
D�
}tjtj|	j�g
|�}|
j|_|
j|�

q�q�WdS)Nr�body�innerrrc
Ssg|]}
|
j��qSr)
r)�.0�
crrr�
<listcomp>�s	z3FixTupleParams.transform_lambda.<locals>.<listcomp>)�
simplify_argsrr�NAMErrr �find_params�map_to_indexr�
tuple_namerZ
post_orderr+rr
r
Zpower)
r'r0r1rr8r9ZparamsZto_indexZtup_nameZ	new_paramr#Z
subscripts�newrrrr)ns(






	







!




zFixTupleParams.transform_lambdaN)�__name__�
__module__�__qualname__Z	run_orderZ
BM_compatibleZPATTERNr7r)rrrrrs


@rc

Cso|jt
jtjfkr|S|jt
jkr[x#|jt
jkrV|jd
}q4W|Std|�
�
dS)NrzReceived unexpected node %s)rr
Zvfplistrr>�vfpdefr�RuntimeError)
r0rrrr=�s





r=c

CsP|jt
jkr#t|jd
�
S|jtjkr<|jSdd�|jD�
S)Nrc
Ss.g|]$}
|
jt
jkrt|
�
�qSr)rr�COMMAr?)r:r;rrrr<�s	zfind_params.<locals>.<listcomp>)rr
rFr?rrr>r+)
r0rrrr?�s





r?NcCs�|dkri}nxht|�
D]Z\}}t
tt|�
�
�
g
}t|t�rnt||d
|�

q"|
|||<q"W|S)N�
d)r,r	r�strr�listr@)�
param_listrrIr4�objZtrailerrrrr@�s

	




r@c
CsVg}
x@|D]8}t|t
�r8|
jt|�
�

q
|
j|�

q
Wd
j|
�
S)N�
_)rrKr!rA�join)rL�
lrMrrrrA�s





rA)�__doc__rrZpgen2rrZ
fixer_utilrrrrr	r
rZBaseFixrr=r?r@rArrrr�<module>s


.l