← Back
Editing: fix_tuple_params.cpython-35.pyc
��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)�Assign�Name�Newline�Number� Subscript�symsc C s) t | t j � o( | j d j t j k S)N� )� isinstancer �Node�children�typer �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 | _ q�W� | d j | | � <x= t | d | t � � d � D] } | | d j | _ q�W| d j � d S)N�lambda�suite�argsr � 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_name�clone�prefixr �replace�appendr r r Z simple_stmt)Z tuple_arg� add_prefix�n�argr )�end� new_lines�selfr r � handle_tupleC s z.FixTupleParams.transform.<locals>.handle_tupler" r )�transform_lambdar r r �INDENT�valuer r ZLeafr ZtfpdefZ typedargslist� enumerate�parentr r �range�lenZchanged) r' �node�resultsr r �start�indentr( �ir$ �line�afterr )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 �body�innerr r c S s g | ] } | j � � q Sr )r )�.0�cr 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 subscripts�newr 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)N�d)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
Save File
Cancel