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.<locals>.handle_tupleC s T]]__%%A//##CCJ#qwwyy))D
a V[)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. sM w ((w77 7 v 8 Q $ 4 4E1X&q)/F))CCEF+elB//C
? ? ? ? ? ? ? ? 9 # # L
Y$,
,
,#DM22
:
:38t{** !L!a%9999 F # #D (DKK A::"%IaL
%(+E2
3
3 "(IaL AIE # #D (DKK)2a%+&uQwc)nn 4Q 677 1 1A*0E!H a ' '
ar c |d }|d }t |d }|j t j k r2| }d|_ | | d S t | }t | }| t | }t |d } | | | D ]}
|
j t j k rv|
j
|v rmd ||
j
D } t j t j | g| z } |
j | _ |
| d S )Nr bodyinnerr! )r$ c 6 g | ]}| S )r# .0cs r
<listcomp>z3FixTupleParams.transform_lambda.<locals>.<listcomp> s C C CAaggii C C Cr )
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*
subscriptsnews
r r0 z FixTupleParams.transform_lambdan s` vv gg .// : # #KKMME EL LL F T"" ''==F!3!344#... Y__&&''' "" Av##8(;(; C C!'1B C C C
k$*#,??#4#4"5
"BD D X
#
r N)__name__
__module__ __qualname__ run_order
BM_compatiblePATTERNrB r0 rG r r r r sD IMG> > >@ r r c | j t j t j fv r| S | j t j k r9| j t j k r"| j d } | j t j k "| S t d| z )Nr zReceived unexpected node %s)r r vfplistr rM vfpdefr RuntimeErrorr; s r rL rL ss yT\5:...
dk ! ! i4;&&=#D i4;&&
4t;
<
<<r c | j t j k rt | j d S | j t
j k r| j S d | j D S )Nr c V g | ]&}|j t j k t | 'S rG )r r COMMArN rH s r rK z find_params.<locals>.<listcomp> s, K K KqQVu{5J5JKNN5J5J5Jr )r r ra rN r r rM r2 rc s r rN rN sS yDK 4=+,,,
ej z K KDM K K KKr Nc |i }t | D ]_\ }}t t t | g}t |t
rt
||| W||z ||< `|S )N)d)r6 r
r strr listrO )
param_listr$ rh r? objtrailers r rO rO s y J'' & &3 VCFF^^,,- c4 & g + + + + +g%AcFF Hr c g }| D ]O}t |t r#| t | :| | Pd | S )N_)r rj r& rP join)rk lrl s r rP rP sd
A c4
HHZ__ % % % %
HHSMMMM 88A;;r )__doc__r r pgen2r r
fixer_utilr r r r r
r r BaseFixr rL rN rO rP rG r r <module>rv s * G G G G G G G G G G G G G G G G1 1 1g g g g g Z' g g g X = = =L L L %'$
r