��Yf~ �@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 Cs�|j}|d}|d}|jd�}|rz|j|jjkrKdS|j|jjkrz|jdjdkrzdS|r�|j|jjkr�|jdjdkr�dS|j}|j�}|jt j |jfkr|j|jks |jdjt j krt|�}d|_|j�}d|_|dk rU|j�}d|_tjt jd�|g}|dk r�|jt�tjt j d�|g�d |d_t||d |�S) N�func�args�kwds�z**r��*� �prefix��r)�syms�get�typeZ star_exprZargumentZchildren�valuerZcloner�NAMEZatomZpower� DOUBLESTARrrZLeaf�STAR�extendrr) �selfZnodeZresultsrr r rrZ l_newargs�r�s