U
e5d @ 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 |st |d}|r4|ttdg |jd d S |jd tdksJt |jdd }t|dkrvt |d rvd S d } }}|r|d t
kr|d d }d}|r|d t t
jdkrt|d kst |d }|d
d }d d |D }|rd
|d _|d k s"|d k s"|d k rz|d k rB| |dtt| |d k rb| |dtt| |d k rz| |d| ttd|} |j| _| S )NZbareprint)prefix z>>r c S s g | ] }| qS )clone).0argr r //usr/lib64/python3.8/lib2to3/fixes/fix_print.py
? s z&FixPrint.transform.. sependfile)AssertionErrorgetreplacer r r Zchildrenlen parend_exprmatchr r Leafr
RIGHTSHIFTr add_kwargr repr)
selfZnodeZresultsZ
bare_printargsr r r Zl_argsZn_stmtr r r transform% s@
zFixPrint.transformc C sN d|_ t| jjt|ttjd|f}|r@| t
d|_ | | d S )Nr =r ) r r ZNodeZsymsZargumentr r" r EQUALappendr )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 s