File: //usr/lib/python3.7/lib2to3/fixes/__pycache__/fix_tuple_params.cpython-37.pyc
B
5a @ s d Z ddlmZ ddlmZ ddlmZ ddlmZmZm Z m
Z
m Z m Z dd Z
G dd d ejZd
d Zd d
Zg dfddZdd ZdS )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 C s t | tjo | jd jtjkS )N )
isinstancer Nodechildrentyper STRING)stmt r 4/usr/lib/python3.7/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|kr ||S g |d }|d }|d jd jtjkrZd}|d jd j}t nd}d}ttjd d fd
d }|jt j
kr|| n@|jt j krx2t |jD ]$\}} | jt j
kr|| |dkd qW sd S xD ]}
|d |
_
qW |} |dkrd
d _n&t|d j| r8|d _|d } xD ]}
|d |
_
q>W |d j| | < x4t| d | t d D ]}||d j| _qW |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 )end new_linesselfr r handle_tupleC s
z.FixTupleParams.transform.<locals>.handle_tuple)r" r )F)transform_lambdar r r INDENTvaluer r ZLeafr
ZtfpdefZ
typedargslist enumerateparentr r rangelenZchanged) r' noderesultsr r startindentr( ir$ lineZafterr )r% r& r' r transform. sF
zFixTupleParams.transformc
C s |d }|d }t |d }|jtjkrD| }d|_|| d S t|}t|}| t
|}t |dd} || xd| D ]X}
|
jtjkr|
j
|krdd ||
j
D } ttj| g| } |
j| _|
| qW d S )Nr bodyinnerr )r c S s g | ] }| qS r )r ).0cr r r
<listcomp> s z3FixTupleParams.transform_lambda.<locals>.<listcomp>)
simplify_argsr r NAMEr r r find_params map_to_indexr
tuple_namer Z
post_orderr+ r r
r
Zpower)
r' r0 r1 r r7 r8 ZparamsZto_indexZtup_nameZ new_paramr# Z
subscriptsnewr r r r) n s(
z FixTupleParams.transform_lambdaN)__name__
__module__ __qualname__Z run_orderZ
BM_compatibleZPATTERNr6 r) r r r r r s
@r c C sR | j tjtjfkr| S | j tjkrBx| j tjkr<| jd } q$W | S td| d S )Nr zReceived unexpected node %s)r r
Zvfplistr r= vfpdefr RuntimeError)r0 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 COMMAr> )r9 r: r r r r; s z find_params.<locals>.<listcomp>)r r
rE r> r r r= r+ )r0 r r r r> s
r> Nc C s^ |d kr i }xLt | D ]@\}}ttt|g}t|trJt|||d q|| ||< qW |S )N)d)r, r r strr listr? )
param_listr rH r4 objZtrailerr r r r? s
r? c C s@ g }x0| D ](}t |tr(|t| q
|| q
W d|S )N_)r rJ r! r@ join)rK lrL r r r r@ s
r@ )__doc__r r Zpgen2r r Z
fixer_utilr r r r r r
r ZBaseFixr r< r> r? r@ r r r r <module> s l