$53 GRAYBYTE WORDPRESS FILE MANAGER $31

a

XC?h��@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
�Zgdfdd�
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

Cst|t
j�o|jd
jtjkS)N�)�
isinstancer�Node�children�typer�STRING)
�stmt�r�C/opt/alt/python39/lib64/python3.9/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s�
d
|vr��|
|�Sg�
|d}|d}|dj
djtjkrZd}|dj
dj}t��nd}d}t�tjd��d��
�fd
d�	}|jt	j
kr�||�

n<|jt	jkr�t|j
�
D]$\}}	|	jt	j
kr�||	|dkd�
q��
s�dS�
D]}
|d|
_
q�|}|dk�
r
d
�
d_n&t|dj
|�
�
r0|�
d_|d}�
D]}
|d|
_
�
q4�
|dj
||�<t|d|t�
�
d�D]}||dj
|_�
qr|d��
dS)N�lambda�suite�argsr�
r
z; �Fcs\t��
��
}|��}d
|_t||���}|
r2d|_|�|�

�
�t�t	j
|���g��

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









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



























zFixTupleParams.transformc
Cs�|d
}|d}t|d�
}|j
tjkrD|��}d|_|�|�

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

|��D]X}
|
j
tjkr�|
j
|vr�dd�||
j
D�
}t�tj|	��g
|�}|
j|_|
�|�

q�dS)Nr�body�innerr)
rc
Ssg|]}
|
���qSr)
r��.0�
crrr�
<listcomp>��z3FixTupleParams.transform_lambda.<locals>.<listcomp>)�
simplify_argsrr�NAMErrr �find_params�map_to_indexr�
tuple_namerZ
post_orderr+rr
r
Zpower)
r'r0r1rr7r8�paramsZto_indexZtup_nameZ	new_paramr#Z
subscripts�newrrrr)ns*















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


@rc

CsN|jt
jtjfvr|S|jt
jkr>|jt
jkr:|jd
}q"|Std|�
�
dS)NrzReceived unexpected node %s)rr
Zvfplistrr?�vfpdefr�RuntimeError�
r0rrrr>�s





r>c

Cs<|jt
jkrt|jd
�
S|jtjkr,|jSdd�|jD�
S)Nrc
Ss g|]}
|
jt
jkrt|
�
�qSr)rr�COMMAr@r9rrrr<�r=zfind_params.<locals>.<listcomp>)rr
rHr@rrr?r+rJrrrr@�s





r@NcCsZ|duri}t|�
D]@\}}t
tt|�
�
�
g
}t|t�rHt|||d
�
q|
|||<q|S)N)
�
d)r,r	r�strr�listrA)�
param_listrrLr4�objZtrailerrrrrA�s







rAc
Cs<g}
|D](}t|t
�r&|
�t|�
�

q|
�|�

qd
�|
�
S)N�
_)rrNr!rB�join)rO�
lrPrrrrB�s





rB)�__doc__rrZpgen2rrZ
fixer_utilrrrrr	r
rZBaseFixrr>r@rArBrrrr�<module>
s


 l