� j fm�@sodZddlmZddlmZddlmZddlmZmZm Z Gdd�dej �ZdS) zIFixer for apply(). This converts apply(func, v, k) into (func)(*v, **k).�)�pytree)�token)� fixer_base)�Call�Comma�parenthesizec@s(eZdZdZdZdd�ZdS)�FixApplyTa. power< 'apply' trailer< '(' arglist< (not argument ')' > > c CsV|j}|st�|d}|d}|jd�}|j}|j�}|jtj|jfkr�|j|j ks�|j d jtjkr�t|�}nd|_|j�}d|_|dk r�|j�}d|_nt jtjd�|g}|dk rC|jt�t jtjd�|g�d|d_nt||d |�S)N�func�args�kwdsr��*z**� �prefix��r)�syms�AssertionError�getrZclone�typer�NAMEZatomZpowerZchildren� DOUBLESTARrrZLeaf�STAR�extendrr) �selfZnodeZresultsrr r rrZ l_newargs�r�s