3 \��@s�ddlmZddlmZddlmZddlmZmZmZm Z m Z mZdd�ZGdd�dej �Zd d �Zdd�Zgd fdd�Zdd�Zd S)�)�pytree)�token)� fixer_base)�Assign�Name�Newline�Number� Subscript�symscCst|tj�o|jdjtjkS)N�)� isinstancer�Node�children�typer�STRING)�stmt�r�6/usr/lib64/python3.6/lib2to3/fixes/fix_tuple_params.py�is_docstringsrc@s(eZdZdZdZdZdd�Zdd�ZdS) �FixTupleParams�Ta funcdef< 'def' any parameters< '(' args=any ')' > ['->' any] ':' suite=any+ > | lambda= lambdef< 'lambda' args=vfpdef< '(' inner=any ')' > ':' body=any > cs�d|kr�j||�Sg�|d}|d}|djdjtjkrZd}|djdj}t��nd}d}tjtjd��d��fd d� }|jt j kr�||�n@|jt jkr�x2t|j�D]$\}} | jt j kr�|| |dkd�q�W�s�dSx�D]} |d| _ q�W|}|dk�rd �d_n&t|dj|��r8|�d_|d}x�D]} |d| _ �q>W�|dj||�<x4t|d|t��d�D]}||dj|_�q�W|dj�dS)N�lambda�suite�argsr�rz; �Fcs\t�j��}|j�}d|_t||j��}|r2d|_|j|��jtjt j |�j�g��dS)Nr� )r�new_name�clone�prefixr�replace�appendrr r Zsimple_stmt)Z tuple_arg� add_prefix�n�argr)�end� new_lines�selfrr�handle_tupleCs z.FixTupleParams.transform..handle_tuple)r"r)F)�transform_lambdarrr�INDENT�valuerrZLeafr ZtfpdefZ typedargslist� enumerate�parentrr�range�lenZchanged)r'�node�resultsrr�start�indentr(�ir$�line�afterr)r%r&r'r� transform.sF zFixTupleParams.transformc Cs�|d}|d}t|d�}|jtjkrD|j�}d|_|j|�dSt|�}t|�}|j t |��}t|dd�} |j| j��xd|j�D]X} | jtjkr�| j |kr�dd�|| j D�}tjtj| j�g|�}| j|_| j|�q�WdS)Nr�body�innerr)rcSsg|]}|j��qSr)r)�.0�crrr� �sz3FixTupleParams.transform_lambda..)� simplify_argsrr�NAMErrr �find_params�map_to_indexr� tuple_namerZ post_orderr+rr r Zpower) r'r0r1rr8r9ZparamsZto_indexZtup_nameZ new_paramr#Z subscripts�newrrrr)ns( zFixTupleParams.transform_lambdaN)�__name__� __module__�__qualname__Z run_orderZ BM_compatibleZPATTERNr7r)rrrrrs @rcCsR|jtjtjfkr|S|jtjkrBx|jtjkr<|jd}q$W|Std|��dS)NrzReceived unexpected node %s)rr Zvfplistrr>�vfpdefr�RuntimeError)r0rrrr=�sr=cCs<|jtjkrt|jd�S|jtjkr,|jSdd�|jD�S)NrcSs g|]}|jtjkrt|��qSr)rr�COMMAr?)r:r;rrrr<�szfind_params..)rr rFr?rrr>r+)r0rrrr?�s r?NcCs^|dkri}xLt|�D]@\}}ttt|��g}t|t�rJt|||d�q||||<qW|S)N)�d)r,r r�strr�listr@)� param_listrrIr4�objZtrailerrrrr@�s r@cCs@g}x0|D](}t|t�r(|jt|��q |j|�q Wdj|�S)N�_)rrKr!rA�join)rL�lrMrrrrA�s rA)rrZpgen2rrZ fixer_utilrrrrr r rZBaseFixrr=r?r@rArrrr�s l