Yf @ s d Z d d l m Z d d l m Z d d l m Z d d l m Z m Z m Z m
Z
m Z m Z d d Z
Gd d d e j Z d
d Z d d
Z g d d d Z d d Z d 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 C s) t | t j o( | j d j t j k S)N )
isinstancer Nodechildrentyper STRING)stmt r C/opt/alt/python35/lib64/python3.5/lib2to3/fixes/fix_tuple_params.py is_docstring s r c @ s: e Z d Z d Z d Z d Z d d Z d d Z d S) FixTupleParams Ta
funcdef< 'def' any parameters< '(' args=any ')' >
['->' any] ':' suite=any+ >
|
lambda=
lambdef< 'lambda' args=vfpdef< '(' inner=any ')' >
':' body=any
>
c s- d | k r j | | Sg | d } | d } | d j d j t j k ry d } | d j d j } t n! d } d } t j t j d d f d
d } | j t j
k r | | nZ | j t j k r.xE t | j D]4 \ } } | j t j
k r | | d | d k q W s8d Sx D] }
| d |
_
q?W| } | d k rxd
d _ n. t | d j | r| d _ | d } x D] }
| d |
_
qW | d j | | <x= t | d | t d D] } | | d j | _ qW| d j d S)Nlambdasuiteargsr r z; Fc s t j } | j } d | _ t | | j } | rK d | _ | j | j t j t j
| 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_tupler" r )transform_lambdar r r INDENTvaluer r ZLeafr
ZtfpdefZ
typedargslist enumerateparentr r rangelenZchanged) r' noderesultsr r startindentr( ir$ lineafterr )r% r& r' r transform. sF
(zFixTupleParams.transformc
C s? | d } | d } t | d } | j t j k r\ | j } d | _ | j | d St | } t | } | j t
| } t | d d } | j | j x | j D]| }
|
j t j k r |
j
| k r d d | |
j
D } t j t j | j g | } |
j | _ |
j | q Wd S)Nr bodyinnerr r c S s g | ] } | j q Sr )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 r8 r9 ZparamsZto_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 so | j t j t j f k r | S| j t j k r[ x# | j t j k rV | j d } q4 W| St d | d S)Nr zReceived unexpected node %s)r r
Zvfplistr r> vfpdefr RuntimeError)r0 r r r r= s r= c C sP | j t j k r# t | j d S| j t j k r< | j Sd d | j D S)Nr c S s. g | ]$ } | j t j k r t | q Sr )r r COMMAr? )r: r; r r r r< s z find_params.<locals>.<listcomp>)r r
rF r? r r r> r+ )r0 r r r r? s
r? Nc C s | d k r i } xh t | D]Z \ } } t t t | g } t | t rk t | | d | q | | | | <q W| S)Nd)r, r r strr listr@ )
param_listr rI r4 objZtrailerr r r r@ s r@ c C sV g } x@ | D]8 } t | t r8 | j t | q
| j | q
Wd j | S)N_)r rK r! rA join)rL lrM r r r rA s
rA )__doc__r r Zpgen2r r Z
fixer_utilr r r r r r
r ZBaseFixr r= r? r@ rA r r r r <module> s . l