a
    X%bg                     @   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 J d}t| }||k r| | }|j}|j tjkrX|d7 }q,|durn|j tjkrnq|j}|j}|dur|j tj	kr|j}|d7 }nd}||fV  |d7 }q,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lenZprev_siblingNAME
DOUBLESTARvalueZnext_siblingEQUAL)
raw_paramsZcurr_idxZmax_idxZ	curr_itemZ	prev_itemnameZnxtdefault_value r   n/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/libpasteurize/fixes/fix_kwargs.py
gen_params   s&     

r   c                 C   s   | d j tjksJ | d j tjkrJ| d   | d   | dd }n| dd }|D ] }|j tjkrt|  qZ dS qZd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   COMMAremover   )r   kwargs_defaultZ	kw_paramsparamr   r   r   remove_params+   s    
r   c                 C   s~   d}d}| dd D ]\}|j tjkr(qq|j tjkr>|s>d}q|j tjkr`|r`|rX|jnd  S |j tjkrd}q|rz|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 }t |D ]$\}}|jtjkr||d  } q:qd S t|}|sJd S t| |jd }|jd }t|}	t|D ]\}
}|d u r|	dt
  |	dtt|
|d |	d qv|	dt
  |	dtt|
|d |	d |	dt
  |	dttt|
|d |
|d |	d qv|	|_d|jd _t|}|r|d	 }t|jd
krr|jd jtjkrr|t  |tdd |t| d S )Nparams   r   )r   kwargs)prefix)r   default)Zassignr   r'   r    arglistr    )	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PATTERNr8   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
   r/   r1   r0   Z_kwargs_default_namer   r   r"   ZBaseFixr#   r   r   r   r   <module>   s   