3
‡Œðig  ã               @   s†   d Z ddlmZ ddlmZmZmZmZmZ ddl	m
Z
mZmZ dZdZdZdZd	d
„ Zefdd„Zefdd„ZG dd„ dejƒZdS )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 tjkrt| ƒdks t‚d}t| ƒ}xŒ||k r¸| | }|j}|j tjkrZ|d7 }q.|dk	rp|j tjkrpP |j}|j	}|dk	r |j tj
kr |j	}|d7 }nd}||fV  |d7 }q.W dS )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_paramsZcurr_idxZmax_idxZ	curr_itemZ	prev_itemÚnameZnxtÚdefault_value© r   ú@/tmp/pip-build-6m2fasei/future/libpasteurize/fixes/fix_kwargs.pyÚ
gen_params   s&     


r   c             C   s†   | d j tjkst‚| d j tjkrJ| d jƒ  | d jƒ  | dd… }n| dd… }x*|D ]}|j tjkrv|jƒ  q\dS q\W dS dS )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| dd… D ]Z}|j tjkr*qq|j tjkrB| rBd}q|j tjkr`|r`|r\|jS dS |j tjkrd}qW |r||S dS dS )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 Zd ZdZdZdd„ ZdS )Ú	FixKwargsé   z^funcdef< 'def' NAME parameters< '(' arglist=typedargslist< params=any* > ')' > ':' suite=any >c          	   C   sš  |d }x2t |ƒD ]"\}}|jtjkr||d … }P qW d S t|ƒ}|sLd S t|ƒ |jd }|jd }t|ƒ}	x®t|ƒD ]¢\}
}|d kr¸|j	dt
ƒ ƒ |j	dtt|
|dœ |	d�ƒ qz|j	dt
ƒ ƒ |j	dtt|
|dœ |	d�ƒ |j	dt
ƒ ƒ |j	dttt|
|dœ |
|dœ |	d�ƒ qzW |	|_d|jd _t|ƒ}|�r–|d	 }t|jƒd
k�rx|jd jtjk�rx|jtƒ ƒ |jtdd�ƒ |jt|ƒƒ d S )NÚparamsé   r   )r   Úkwargs)Úprefix)r   Údefault)Zassignr   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ÚresultsZparams_rawlistÚiÚitemZ
new_kwargsZsuiteZ
first_stmtÚidentr   r   Zmust_add_kwargsr+   r   r   r   Ú	transform`   s<    

 .$zFixKwargs.transformN)Ú__name__Ú
__module__Ú__qualname__Z	run_orderZPATTERNr9   r   r   r   r   r$   Z   s   r$   N)Ú__doc__Zlib2to3r   Zlib2to3.fixer_utilr   r   r   r   r   Zlibfuturize.fixer_utilr   r	   r
   r0   r2   r1   Z_kwargs_default_namer   r    r#   ZBaseFixr$   r   r   r   r   Ú<module>   s   