3
+��hg � @ s� d Z ddlmZ ddlmZmZmZmZmZ ddl m
Z
m Z m Z dZ
dZdZdZd d
� Zefd d �Zefd
d�ZG dd� dej�ZdS )zg
Fixer for Python 3 function parameter syntax
This fixer is rather sensitive to incorrect py3k syntax.
� )�
fixer_base)�token�String�Newline�Comma�Name)� indentation�suitify�
DoubleStarz=%(name)s = %(kwargs)s['%(name)s']; del %(kwargs)s['%(name)s']z'if '%(name)s' in %(kwargs)s: %(assign)sz else: %(name)s = %(default)sZ _3to2kwargsc c s� | d j tjkr t| �dks t�d}t| �}x�||k r�| | }|j}|j tjkrZ|d7 }q.|dk rp|j tjkrpP |j}|j }|dk r�|j tj
kr�|j }|d7 }nd}||fV |d7 }q.W dS )z�
Generator that yields tuples of (name, default_value) for each parameter in the list
If no default is given, then it is default_value is None (not Leaf(token.NAME, 'None'))
r � � N) �typer �STAR�len�AssertionErrorZ prev_sibling�NAME�
DOUBLESTAR�valueZ next_sibling�EQUAL)�
raw_paramsZcurr_idxZmax_idxZ curr_itemZ prev_item�nameZnxt�
default_value� r �E/root/tmp/pip-build-yeh9mt8d/future/libpasteurize/fixes/fix_kwargs.py�
gen_params s&
r c C s� | d j tjkst�| d j tjkrJ| d j� | d j� | dd� }n | dd� }x*|D ] }|j tjkrv|j� q\dS q\W dS dS )z�
Removes all keyword-only args from the params list and a bare star, if any.
Does not add the kwargs dict if needed.
Returns True if more action is needed, False if not
(more action is needed if no kwargs dict exists)
r r r N� FT)r
r r r �COMMA�remover )r �kwargs_defaultZ kw_params�paramr r r �
remove_params+ s
r c C s� d}d}xv| dd� D ]Z}|j tjkr*qq|j tjkrB| rBd}q|j tjkr`|r`|r\|jS dS |j tjkrd}qW |r||S dS dS )z�
Returns string with the name of the kwargs dict if the params after the first star need fixing
Otherwise returns empty string
Fr NT� )r
r r r r r )r r Z found_kwargsZ needs_fix�tr r r � needs_fixingA s r# c @ s e Zd ZdZdZdd� ZdS )� FixKwargs� z^funcdef< 'def' NAME parameters< '(' arglist=typedargslist< params=any* > ')' > ':' suite=any >c C s� |d }x2t |�D ]"\}}|jtjkr||d � }P qW d S t|�}|sLd S t|� |jd }|jd }t|�} x�t|�D ]�\}
} | d kr�|j dt
� � |j dt t |
|d� | d�� qz|j dt
� � |j dt t
|
| d� | d�� |j dt
� � |j dt tt |
|d� |
|d� | d�� qzW | |_d|jd _t|�} | �r�|d }
t|
j�d
k�rx|
jd
jtjk�rx|
jt� � |
jtd d�� |
jt|�� d S )N�params� r )r �kwargs)�prefix)r �default)Zassignr r( r! �arglistr r � ���)� enumerater
r r r# r �childrenr r Z insert_childr r �_assign_template�_else_template� _if_templater) r r r Z append_childr r
r )�self�node�resultsZparams_rawlist�i�itemZ
new_kwargsZsuiteZ
first_stmt�identr r Zmust_add_kwargsr+ r r r � transform` s<
. $ zFixKwargs.transformN)�__name__�
__module__� __qualname__Z run_orderZPATTERNr9 r r r r r$ Z s r$ N)�__doc__Zlib2to3r Zlib2to3.fixer_utilr r r r r Zlibfuturize.fixer_utilr r r
r0 r2 r1 Z_kwargs_default_namer r r# ZBaseFixr$ r r r r �<module> s |