3

�cg�@s�dZd
dl
mZ
d
dlmZmZmZmZmZ
d
dl	m
Z
mZmZ
dZ
dZdZdZd	d
�Zef
dd�
Zef
d
d�
ZGdd�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)szelse: %(name)s = %(default)sZ_3to2kwargsc
cs�d
}
t|�
}x�|
|kr�||
}|j
}|jtjkr:|
d7}
q|dk	rP|jtjkrPP|j}|j}|dk	r�|jtjkr�|j}|
d
7}
nd}||fV
|
d7}
qWdS)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'))
    ��
N)	�lenZprev_sibling�typer�NAME�
DOUBLESTAR�valueZnext_sibling�EQUAL)�
raw_paramsZcurr_idxZmax_idxZ	curr_itemZ	prev_item�nameZnxt�
default_value�r� /usr/lib/python3.6/fix_kwargs.py�
gen_paramss$


















rcCsr|d
jt
jkr6|dj�
|d
j�
|dd�}n|dd�}x*|D]}|jt
jkrb|j�
qHdSqHWdSdS)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
rN�FT)rr�COMMA�remover)r�kwargs_defaultZ	kw_paramsZparamrrr�
remove_params+s







rcCs�d
}d
}xv|dd�D]Z}|jt
jkr*qq|jt
jkrB|rBd}q|jt
jkr`|r`|r\|jSdS|jt
jkrd}qW|r||
SdSdS)z�
    Returns string with the name of the kwargs dict if the params after the first star need fixing
    Otherwise returns empty string
    FrNT�)rrrrrr)rrZfound_kwargsZ	needs_fix�
trrr�needs_fixingAs




r c@seZ
dZd
ZdZdd�ZdS)�	FixKwargs�z^funcdef< 'def' NAME parameters< '(' arglist=typedargslist< params=any* > ')' > ':' suite=any >c	Cs�
|d
}x2t|�
D]"\}}|j
tjkr||d�}PqWdSt|�
}|sLdSt|
�

|
jd}|jd}t|�
}	x�t|�
D]�\}
}|dkr�|j	dt
��
|j	dtt|
|d�|	d��
qz|j	dt
��
|j	dtt
|
|d�|	d��
|j	dt
��
|j	dttt|
|d�|
|d�|	d��
qzW|	|_d|jd_t|�
}|�
r�|d	}
t|
j�
d
k�
rx|
jd
j
tjk�
rx|
jt��

|
jtdd�
�

|
jt|�
�

dS)NZparams�r)r�kwargs)
�prefix)r�default)Zassignrr$r�arglistr
r�
 ���)�	enumeraterr�STARr r	ZchildrenrrZinsert_childrr�_assign_template�_else_template�_if_templater%rr
rZappend_childrr
r)�selfZnode�resultsZparams_rawlist�
i�itemZ
new_kwargsZsuiteZ
first_stmtZidentrrZmust_add_kwargsr'rrr�	transform`s<













 


.




$


zFixKwargs.transformN)�__name__�
__module__�__qualname__Z	run_orderZPATTERNr3rrrrr!Zsr!N)�__doc__Zlib2to3rZlib2to3.fixer_utilrrrrrZlibfuturize.fixer_utilrr	r
r,r.r-Z_kwargs_default_namerrr ZBaseFixr!rrrr�<module>s