a DOg @sldZddlmZddlmZddlmZddlmZddlmZm Z m Z m Z e dZ Gdd 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$eZdZdZdZddZddZdS)FixPrintTzP simple_stmt< any* bare='print' any* > | print_stmt c Cs|sJ|d}|r4|ttdg|jddS|jdtdksJJ|jdd}t|dkrvt|drvdSd}}}|r|dt kr|dd}d}|r|dt t j dkrt|d ksJ|d}|d d}d d |D}|rd |d_|dus"|dus"|durz|durB||dtt||durb||dtt||durz||d|ttd|} |j| _| S)NZbareprint)prefix z>>rcSsg|] }|qS)clone).0argrr?z&FixPrint.transform..sependfile)getreplacerrr Zchildrenlen parend_exprmatchrrLeafr RIGHTSHIFTr add_kwargr repr) selfZnodeZresultsZ bare_printargsrrrZl_argsZn_stmtrrr transform%s@         zFixPrint.transformcCsNd|_t|jjt|ttjd|f}|r@| t d|_| |dS)Nr=r) r rZNodeZsymsZargumentrr"rEQUALappendr)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 Zcompile_patternr ZBaseFixr rrrrs