B �A�[8 �@s\dZddlmZmZmZddlmZddlmZm Z m Z mZe�d�Z Gdd�dej�ZdS) a,Fixer for print. Change: "print" into "print()" "print ..." into "print(...)" "print(...)" not changed "print ... ," into "print(..., end=' ')" "print >>x, ..." into "print(..., file=x)" No changes are applied if print_function is imported from __future__ �)�patcomp�pytree� fixer_base)�token)�Name�Call�Comma�Stringz8atom< '(' [arith_expr|atom|power|term|STRING|NAME] ')' >c@s$eZdZdZdZdd�Zdd�ZdS)�FixPrintTzP simple_stmt< any* bare='print' any* > | print_stmt c Cs�|st�|�d�}|r4|�ttd�g|jd��dS|jdtd�ksJt�|jdd�}t|�dkrvt� |d�rvdSd}}}|r�|dt �kr�|dd�}d}|r�|dt�t jd�kr�t|�d ks�t�|d��}|d d�}dd�|D�}|�rd |d_|dk �s"|dk �s"|dk �rz|dk �rB|�|dtt|��|dk �rb|�|dtt|��|dk �rz|�|d|�ttd�|�} |j| _| S)NZbare�print)�prefixr�� z>>��cSsg|]}|��qS�)�clone)�.0�argrr�L/opt/alt/python37/lib/python3.7/site-packages/libfuturize/fixes/fix_print.py� Asz&FixPrint.transform..��sep�end�file)�AssertionError�get�replacerrrZchildren�len�parend_expr�matchrr�Leafr� RIGHTSHIFTr� add_kwargr �repr) �selfZnodeZresultsZ bare_print�argsrrrZl_argsZn_stmtrrr� transform$s> zFixPrint.transformcCsNd|_t�|jjt|�t�tjd�|f�}|r@|� t ��d|_|� |�dS)Nr�=r)rrZNodeZsymsZargumentrr"r�EQUAL�appendr)r&Zl_nodesZs_kwdZn_exprZ n_argumentrrrr$Ts zFixPrint.add_kwargN)�__name__� __module__�__qualname__Z BM_compatibleZPATTERNr(r$rrrrr s0r N)�__doc__Zlib2to3rrrZ lib2to3.pgen2rZlib2to3.fixer_utilrrrr Zcompile_patternr ZBaseFixr rrrr�s