î i f& ã@sšdZddlmZddlmZddlmZddlmZddlmZm Z m Z m Z m Z ej dƒZGdd „d ejƒZd 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@s4eZdZdZdZdd„Zdd„ZdS)ÚFixPrintTzP simple_stmt< any* bare='print' any* > | print_stmt c Csõ|jdƒ}|r>|jttdƒgd|jƒƒdS|jdd…}t|ƒdkrztj|dƒrzdSd}}}|rº|dt ƒkrº|dd…}d}n|r|dt j t j dƒkr|djƒ}|dd…}nd d „|Dƒ}|r+d |d_n|dk sO|dk sO|dk rÐ|dk r}|j|d tt|ƒƒƒn|dk r«|j|d tt|ƒƒƒn|dk rÐ|j|d|ƒqÐnttdƒ|ƒ} |j| _| S)NZbareÚprintÚprefixééú z>>écSsg|]}|jƒ‘qS©)Úclone)Ú.0Úargrrú?s z&FixPrint.transform..ÚÚsepÚendÚfileéÿÿÿÿr)ÚgetÚreplacerrr ZchildrenÚlenÚ parend_exprÚmatchrrÚLeafrÚ RIGHTSHIFTrÚ add_kwargr Úrepr) ÚselfZnodeZresultsZ bare_printÚargsrrrZl_argsZn_stmtrrrÚ transform%s8 % %$ " "  zFixPrint.transformcCsrd|_tj|jjt|ƒtjtjdƒ|fƒ}|ra|j t ƒƒd|_n|j |ƒdS)Nrú=r) r rZNodeZsymsZargumentrr"rÚEQUALÚappendr)r&Zl_nodesZs_kwdZn_exprZ n_argumentrrrr$Ms    zFixPrint.add_kwargN)Ú__name__Ú __module__Ú __qualname__Z BM_compatibleZPATTERNr(r$rrrrr s  (r N)Ú__doc__rrrZpgen2rrZ fixer_utilrrrr r Zcompile_patternr ZBaseFixr rrrrÚs(