f d Z ddlmZ ddlmZ ddlmZ ddlmZmZm Z m
Z
m Z m Z d Z
G d dej Zd Zd
Zg d fd Zd
Zd S )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)AssignNameNewlineNumber Subscriptsymsc v t | t j o | j d j t
j k S )N )
isinstancer Nodechildrentyper STRING)stmts E/opt/alt/python311/lib64/python3.11/lib2to3/fixes/fix_tuple_params.py is_docstringr s/ dFK ( ( 1 = EL 01 c & e Zd ZdZdZdZd Zd ZdS )FixTupleParams Ta
funcdef< 'def' any parameters< '(' args=any ')' >
['->' any] ':' suite=any+ >
|
lambda=
lambdef< 'lambda' args=vfpdef< '(' inner=any ')' >
':' body=any
>
c
d|v r || S g
|d }|d }|d j d j t j k r)d}|d j d j }t
n#d}d}t j t j d d
fd
}|j t j
k r || nU|j t j k r@t |j D ]+\ }} | j t j
k r || |dk ,
sd S
D ]}
|d |
_
|} |dk rd
d _ n2t |d j | r|
d _ |dz }
D ]}
|d |
_
|d j | | < t! | dz | t#
z dz D ]}||d j | _ |d d S )Nlambdasuiteargsr
r z; Fc t }| }d|_ t || }|rd|_ | | t j t j
| g d S )Nr ) r new_namecloneprefixr replaceappendr r r simple_stmt) tuple_arg
add_prefixnargr end new_linesselfs r handle_tuplez.FixTupleParams.transform.