B
� f* � @ sR d Z ddlmZ ddlmZ ddlmZ ddlmZmZm Z G dd� dej
�Z dS ) zIFixer for apply().
This converts apply(func, v, k) into (func)(*v, **k).� )�pytree)�token)�
fixer_base)�Call�Comma� parenthesizec @ s e Zd ZdZdZdd� ZdS )�FixApplyTa.
power< 'apply'
trailer<
'('
arglist<
(not argument
')'
>
>
c C s, | j }|d }|d }|�d�}|rF|j| j jkrF|jd jdkrFd S |rl|j| j jkrl|jd jdkrld S |j}|�� }|jtj |j
fkr�|j|j ks�|jd jtj kr�t
|�}d|_|�� }d|_|d k r�|�� }d|_t�tjd �|g}|d k �r |�t� t�tj d�|g� d
|d _t|||d �S ) N�func�args�kwds� > �**�*z**������ r � )�prefix)�syms�get�typeZargumentZchildren�valuer Zcloner �NAMEZatomZpower�
DOUBLESTARr r ZLeaf�STAR�extendr r ) �selfZnodeZresultsr r r
r r Z l_newargs� r � s