a
XC?h @ sl d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZm Z m
Z
m Z e dZ
G dd d ejZd
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)NameCallCommaStringz"atom< '(' [atom|STRING|NAME] ')' >c @ s$ e Zd ZdZdZdd Zdd ZdS )FixPrintTzP
simple_stmt< any* bare='print' any* > | print_stmt
c
C s` | d}|r,|ttdg |jd d S |jdd }t|dkrXt|d rXd S d } }}|r|d t kr|d d }d}|r|d t
t j
dkr|d }|d d }d
d |D }|rd |d _|d us|d us|d urF|d ur| |d
tt| |d ur.| |dtt| |d urF| |d| ttd|} |j| _| S )NZbareprint)prefix z>> c S s g | ] }| qS )clone).0argr r ? z&FixPrint.transform.