{|j d Z ddlmZ ddlmZ ddlmZ ddlmZmZm Z m
Z
m Z m Z d Z
G d dej Zd Zd
Zg d fd Zd
Zy )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 t | t j xr* | j d j t
j k( S )N )
isinstancer Nodechildrentyper STRING)stmts 7/usr/lib64/python3.12/lib2to3/fixes/fix_tuple_params.py is_docstringr s5 dFKK ( 1 == ELL 01 c $ e Zd ZdZdZdZd Zd Zy)FixTupleParams Ta
funcdef< 'def' any parameters< '(' args=any ')' >
['->' any] ':' suite=any+ >
|
lambda=
lambdef< 'lambda' args=vfpdef< '(' inner=any ')' >
':' body=any
>
c
d|v r j || 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 || ne|j t j k( rHt |j D ]0 \ }} | j t j k( s$ || |dkD
2
sy
D ] }
|d |
_
|} |dk( r d
d _ n*t |d j | r|
d _ |dz }
D ] }
|d |
_
|d j | | t! | dz | t#
z dz D ] }||d j | _ |d j% y )
Nlambdasuiteargsr
r z; c T t j }| j }d|_ t ||j }|rd|_ | j | j
t j t j |j g y )Nr ) r new_namecloneprefixr replaceappendr r r simple_stmt) tuple_arg
add_prefixnargr end new_linesselfs r handle_tuplez.FixTupleParams.transform.