B âxác ã@sldZddlmZddlmZddlmZddlmZddlmZm Z m Z m Z e  d¡Z Gdd „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ÚStringz"atom< '(' [atom|STRING|NAME] ')' >c@s$eZdZdZdZdd„Zdd„ZdS)ÚFixPrintTzP simple_stmt< any* bare='print' any* > | print_stmt c Cs”|st‚| d¡}|r4| ttdƒg|jd¡dS|jdtdƒksJt‚|jdd…}t|ƒdkrvt  |d¡rvdSd}}}|r¤|dt ƒkr¤|dd…}d}|ræ|dt   t jd¡kræt|ƒd ksÎt‚|d ¡}|d d…}d d „|Dƒ}|rd |d_|dk s"|dk s"|dk rz|dk rB| |dtt|ƒƒ¡|dk rb| |dtt|ƒƒ¡|dk rz| |d|¡ttdƒ|ƒ} |j| _| S)NZbareÚprint)Úprefixéééÿÿÿÿú z>>récSsg|] }| ¡‘qS©)Úclone)Ú.0Úargrrú?sz&FixPrint.transform..ÚÚsepÚendÚfile)ÚAssertionErrorÚgetÚreplacerrr ZchildrenÚlenÚ parend_exprÚmatchrrÚLeafrÚ RIGHTSHIFTrÚ add_kwargr Úrepr) ÚselfZnodeZresultsZ bare_printÚargsrrrZl_argsZn_stmtrrrÚ transform%s>          zFixPrint.transformcCsNd|_t |jjt|ƒt tjd¡|f¡}|r@|  t ƒ¡d|_|  |¡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 Zcompile_patternr ZBaseFixr rrrrÚs