3
\ � @ sl d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZm Z m
Z
m Z ej d�Z
G dd � d ej�Zd
S ) a Fixer for print.
Change:
'print' into 'print()'
'print ...' into 'print(...)'
'print ... ,' into 'print(..., end=" ")'
'print >>x, ...' into 'print(..., file=x)'
No changes are applied if print_function is imported from __future__
� )�patcomp)�pytree)�token)�
fixer_base)�Name�Call�Comma�Stringz"atom< '(' [atom|STRING|NAME] ')' >c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�FixPrintTzP
simple_stmt< any* bare='print' any* > | print_stmt
c
C s� |st �|jd�}|r4|jttd�g |jd�� d S |jd td�ksJt �|jdd � }t|�dkrvtj |d �rvd S d } }}|r�|d t
� kr�|d d� }d}|r�|d t j t
jd�kr�t|�dks�t �|d j� }|d d � }d
d � |D �}|�rd |d _|d k �s"|d k �s"|d k �rz|d k �rB| j|d
tt|��� |d k �rb| j|dtt|��� |d k �rz| j|d|� ttd�|�} |j| _| S )NZbare�print)�prefix� � � z>>r � c S s g | ] }|j � �qS � )�clone)�.0�argr r �//usr/lib64/python3.6/lib2to3/fixes/fix_print.py�
<listcomp>? s z&FixPrint.transform.<locals>.<listcomp>� �sep�end�file���r )�AssertionError�get�replacer r r Zchildren�len� parend_expr�matchr r �Leafr �
RIGHTSHIFTr � add_kwargr �repr)
�selfZnodeZresultsZ
bare_print�argsr r r Zl_argsZn_stmtr r r � transform% s>
zFixPrint.transformc C sN d|_ tj| jjt|�tjtjd�|f�}|r@|j t
� � d|_ |j |� d S )Nr �=r ) r r ZNodeZsymsZargumentr r" r �EQUAL�appendr )r& Zl_nodesZs_kwdZn_exprZ
n_argumentr r r r$ M s
zFixPrint.add_kwargN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr( r$ r r r r r
s (r
N)�__doc__r r r Zpgen2r r Z
fixer_utilr r r r Zcompile_patternr ZBaseFixr
r r r r �<module> s