ðÕYf& ã @ sš d Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z m Z m
Z
m Z m Z e j
d ƒ Z Gd d „ d e j ƒ Z d
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ÚStringÚis_tuplez"atom< '(' [atom|STRING|NAME] ')' >c @ s4 e Z d Z d Z d Z d d „ Z d d „ Z d S)ÚFixPrintTzP
simple_stmt< any* bare='print' any* > | print_stmt
c
C sà | j d ƒ } | r> | j t t d ƒ g d | j ƒƒ d S| j d d … } t | ƒ d k rz t j | d ƒ rz d Sd } } } | r· | d t ƒ k r· | d d … } d } | rü | d t
j t j
d ƒ k rü | d j ƒ } | d d … } d d
„ | Dƒ } | r"d | d _ | d k sF| d k sF| d k r»| d k rq| j | d t t | ƒ ƒ ƒ | d k rœ| j | d
t t | ƒ ƒ ƒ | d k r»| j | d | ƒ t t d ƒ | ƒ } | j | _ | S)NZbareÚprintÚprefixé é ú z>>é c S s g | ] } | j ƒ ‘ q S© )Úclone)Ú.0Úargr r ú? s z&FixPrint.transform..Ú ÚsepÚendÚfileéÿÿÿÿr )Ú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% s8
%%
$ zFixPrint.transformc C so d | _ t j | j j t | ƒ t j t j d ƒ | 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 r
Zcompile_patternr ZBaseFixr r r r r Ú s (