î
i f& ã @ 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 } n | r| d t
j t j
d ƒ k r| d j ƒ } | d d … } n d d
„ | Dƒ } | r+d | d _ n | d k sO| d k sO| d k rÐ| d k r}| j | d t t | ƒ ƒ ƒ n | d k r«| j | d
t t | ƒ ƒ ƒ n | d k rÐ| j | d | ƒ qÐn 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 sr d | _ t j | j j t | ƒ t j t j d ƒ | f ƒ } | ra | j t
ƒ ƒ d | _ n | 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 (