֦i 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..handle_tupleC s T]]_%A//#CCJ#qwwy)D a V[[)9)9*. )< >
?r )r) r! )F)transform_lambdar r r INDENTvaluer r Leafr tfpdef
typedargslist enumerateparentr$ r rangelenchanged)r. noderesultsr r startindentr/ ir+ lineafterr, r- s ` @@r transformzFixTupleParams.transform. s w ((w7 7 v 8 Q $ $ 4E1X&&q)//F)CEF++ellB/C
? 99 #
YY$,,
,#DMM2388t{{* !!a%9 3 D (DK A:"%IaL
%(++E2
3"(IaL AIE D (DK )2a%&uQwc)n 4Q 67A*0E!H a ' 8
ar c
|d }|d }t |d }|j t j k( r)|j }d|_ |j
| y t | }t | }| j t | }t |d } |j
| j |j D ] }
|
j t j k( s!|
j |v s0||
j D cg c] } | j } } t j t j" | j g| z }
|
j
|
_ |
j
|
y c c} w )Nr bodyinnerr! )r$ )
simplify_argsr r NAMEr# r$ r% find_params map_to_indexr"
tuple_namer
post_orderr2 r r r power)r. r; r< r rD rE paramsto_indextup_name new_paramr* c
subscriptsnews r r0 z FixTupleParams.transform_lambdan s- vv gg ./ :: #KKME EL LL T" '==F!34#. Y__&' "Avv#8(;19!''1B C1BAaggi1B
C kk$**#,??#4"5
"BD XX
#
# Cs F N)__name__
__module__ __qualname__ run_order
BM_compatiblePATTERNrB r0 r r r r s IMG>@ r r c L | j t j t j fv r| S | j t j
k( rL| j t j
k( r-| j d } | j t j
k( r-| S t d| z )Nr zReceived unexpected node %s)r r vfplistr rG vfpdefr RuntimeError)r; s r rF rF sx yyT\\5::..
dkk ! ii4;;&==#D ii4;;&
4t;
< }t |t r|j t | .|j | @ dj | S )N_)r rd r& rJ join)re lrf s r rJ rJ sF
A c4
HHZ_ %
HHSM
88A;r )__doc__r r pgen2r r
fixer_utilr r r r r
r r BaseFixr rF rH rI rJ rZ r r rp sQ * G G1g Z'' g X =L %'$
r