o

,�]8
�@s\dZd
dl
mZmZmZ
d
dlmZ
d
dlmZm	Z	m
Z
mZ
e�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$eZ
dZd
ZdZdd�Zdd�ZdS)�FixPrintTzP
              simple_stmt< any* bare='print' any* > | print_stmt
              c
Cs�
|sJ�
|�d
�
}|r|�
ttd�
g|jd��

dS|
jdtd�
ks%J�
|
jdd�}t|�
dkr;t�|d�
r;dSd}}}|rR|dt	�krR|dd�}d}|rs|dt
�tj
d�krst|�
d	ksgJ�
|d��}|d
d�}dd�|D�
}|r�d
|d_|du
s�|du
s�|du
r�|du
r�|�|dtt|�
�
�
|du
r�|�|dtt|�
�
�
|du
r�|�|d|�
ttd�
|�}	|
j|	_|	S)N�bare�print)
�prefixr
�
����
 z>>��c
Ssg|]}
|
���qS�)
�clone)�.0�argrr�=/usr/lib/python3/dist-packages/libfuturize/fixes/fix_print.py�
<listcomp>Asz&FixPrint.transform.<locals>.<listcomp>��sep�end�file)�get�replacerrr
�children�len�parend_expr�matchrr�Leafr�
RIGHTSHIFTr�	add_kwargr	�repr)
�self�node�results�
bare_print�argsrrr�l_args�n_stmtrrr�	transform$s@


�





















zFixPrint.transformcCsNd
|_t
�|jjt|�
t
�tjd�|f�}|
r |
�	t
��

d|_|
�	|�

dS)Nr�
=r)r
r�Node�syms�argumentrr#r�EQUAL�appendr)r'�l_nodes�s_kwd�n_expr�
n_argumentrrrr%Ts




��


zFixPrint.add_kwargN)�__name__�
__module__�__qualname__�
BM_compatible�PATTERNr.r%rrrrr
s
0r
N)�__doc__�lib2to3rrr�
lib2to3.pgen2r�lib2to3.fixer_utilrrrr	�compile_patternr!�BaseFixr
rrrr�<module>
s


�