��]&�@s�ddlmZddlmZddlmZddlmZddlmZmZm Z m Z mZejd�Z Gdd�dej�Zd S) �)�patcomp)�pytree)�token)� fixer_base)�Name�Call�Comma�String�is_tuplez"atom< '(' [atom|STRING|NAME] ')' >c@s4eZdZdZdZdd�Zdd�ZdS)�FixPrintTzP simple_stmt< any* bare='print' any* > | print_stmt c Cs�|jd�}|r>|jttd�gd|j��dS|jdd�}t|�dkrztj|d�rzdSd}}}|r�|dt �kr�|dd�}d}|r�|dt jtj d�kr�|dj�}|dd�}d d �|D�}|r"d|d_|dk sF|dk sF|dk r�|dk rq|j|dtt|��|dk r�|j|d tt|��|dk r�|j|d|�ttd�|�} |j| _| S)NZbare�print�prefix�� z>>�cSsg|]}|j��qS�)�clone)�.0�argrr�./opt/alt/python35/lib64/python3.5/fix_print.py� ?s z&FixPrint.transform..��sep�end�file��r)�get�replacerrr Zchildren�len�parend_expr�matchrr�Leafr� RIGHTSHIFTr� add_kwargr �repr) �selfZnode�resultsZ bare_print�argsrrrZl_argsZn_stmtrrr� transform%s8 %% $zFixPrint.transformcCsod|_tj|jjt|�tjtjd�|f�}|r^|j t ��d|_|j |�dS)Nr�=r)r rZNodeZsyms�argumentrr"r�EQUAL�appendr)r&Zl_nodesZs_kwdZn_exprZ n_argumentrrrr$Ms zFixPrint.add_kwargN)�__name__� __module__�__qualname__Z BM_compatibleZPATTERNr)r$rrrrrs(rN)rrrZpgen2rrZ fixer_utilrrrr r Zcompile_patternr ZBaseFixrrrrr�s(