a
i @ s d dl mZ d dlmZ d dl mZ d dlmZmZmZm Z m
Z
m Z dd Z G dd dej
Zd d
Zd d Zg d
fddZdd Zd
S ) )pytree)token)
fixer_base)AssignNameNewlineNumber Subscriptsymsc C s t | tjo | jd jtjkS )N )
isinstancer Nodechildrentyper STRING)stmt r 6/usr/lib64/python3.9/lib2to3/fixes/fix_tuple_params.py is_docstring s r c @ s( e Zd ZdZdZdZdd 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|v r ||S g |d }|d }|d jd jtjkrZd}|d jd j}t nd}d}ttjd d fd
d }|jt j
kr|| n<|jt j krt |jD ]$\}} | jt j
kr|| |dkd qsd S D ]}
|d |
_
q|} |dkr
d
d _n&t|d j| r0|d _|d } D ]}
|d |
_
q4|d j| | < t| d | t d D ]}||d j| _qr|d d S )Nlambdasuiteargsr r z; Fc s\ t }| }d|_t|| }|r2d|_| | tt j
| g d S )Nr ) r new_namecloneprefixr replaceappendr r
r
Z simple_stmt)Z tuple_arg
add_prefixnargr endZ new_linesselfr r handle_tupleC s
z.FixTupleParams.transform..handle_tuple)r" r )F)transform_lambdar r r INDENTvaluer r ZLeafr
ZtfpdefZ
typedargslist enumerateparentr r rangelenchanged) r' noderesultsr r startindentr( ir$ lineZafterr r% r transform. sF
zFixTupleParams.transformc
C s |d }|d }t |d }|jtjkrD| }d|_|| d S t|}t|}| t
|}t |dd} || | D ]X}
|
jtjkr|
j
|v rdd ||
j
D } ttj| g| } |
j| _|
| qd S )Nr bodyinnerr )r c S s g | ] }| qS r )r .0cr r r
z3FixTupleParams.transform_lambda..)
simplify_argsr r NAMEr r r find_params map_to_indexr
tuple_namer Z
post_orderr+ r r
r
Zpower)
r' r1 r2 r r8 r9 paramsZto_indexZtup_nameZ new_paramr# Z
subscriptsnewr r r r) n s*
z FixTupleParams.transform_lambdaN)__name__
__module__ __qualname__Z run_orderZ
BM_compatibleZPATTERNr7 r) r r r r r s
@r c C sN | j tjtjfv r| S | j tjkr>| j tjkr:| jd } q"| S td| d S )Nr zReceived unexpected node %s)r r
Zvfplistr r@ vfpdefr RuntimeErrorr1 r r r r? s r? c C s< | j tjkrt| jd S | j tjkr,| jS dd | jD S )Nr c S s g | ]}|j tjkrt|qS r )r r COMMArA r: r r r r= r> z find_params..)r r
rI rA r r r@ r+ rK r r r rA s
rA Nc C sZ |d u r i }t | D ]@\}}ttt|g}t|trHt|||d q|| ||< q|S )N)d)r, r r strr listrB )
param_listr rM r5 objZtrailerr r r rB s
rB c C s< g }| D ](}t |tr&|t| q|| qd|S )N_)r rO r! rC join)rP lrQ r r r rC s
rC )r r Zpgen2r r Z
fixer_utilr r r r r r
r ZBaseFixr r? rA rB rC r r r r s l