a
i @ sh d dl mZ d dl mZ d dlmZ d dl mZ d dlmZmZm Z m
Z
e dZ G dd dej
Zd S )
)patcomp)pytree)token)
fixer_base)NameCallCommaStringz"atom< '(' [atom|STRING|NAME] ')' >c @ s$ e Zd ZdZdZdd Zdd ZdS )FixPrintTzP
simple_stmt< any* bare='print' any* > | print_stmt
c
C s` | d}|r,|ttdg |jd d S |jdd }t|dkrXt|d rXd S d } }}|r|d t kr|d d }d}|r|d t
t j
dkr|d }|d d }d
d |D }|rd |d _|d us|d us|d urF|d ur| |d
tt| |d ur.| |dtt| |d urF| |d| ttd|} |j| _| S )NZbareprint)prefix z>> c S s g | ] }| qS )clone).0argr r //usr/lib64/python3.9/lib2to3/fixes/fix_print.py
? z&FixPrint.transform.. sependfile)getreplacer r r Zchildrenlen parend_exprmatchr r Leafr
RIGHTSHIFTr add_kwargr repr)
selfZnoderesultsZ
bare_printargsr r r Zl_argsZn_stmtr r r transform% s:
zFixPrint.transformc C sN d|_ t| jjt|ttjd|f}|r@| t
d|_ | | d S )Nr =r ) r r ZNodeZsymsargumentr r" r EQUALappendr )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)r r r Zpgen2r r Z
fixer_utilr r r r compile_patternr ZBaseFixr
r r r r s