XylotrechusZ
���]& � @ s� 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)
� )�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 } | r� | d t
j t j
d � k r� | d j � } | d d � } d d
� | D� } | r"d | d _ | d k sF| d k sF| d k r�| d k rq| j | d t t | � � � | d k r�| j | d
t t | � � � | d k r�| j | d | � t t d � | � } | j | _ | S)NZbare�print�prefix� � � z>>� c S s g | ] } | j � � q S� )�clone)�.0�argr r �./opt/alt/python35/lib64/python3.5/fix_print.py�
<listcomp>? s z&FixPrint.transform.<locals>.<listcomp>� �sep�end�file���r )�get�replacer r r
Zchildren�len� parend_expr�matchr r �Leafr �
RIGHTSHIFTr � add_kwargr �repr)
�selfZnode�resultsZ
bare_print�argsr r r Zl_argsZn_stmtr r r � transform% s8
%%
$ zFixPrint.transformc C so d | _ t j | j j t | � t j t j d � | f � } | r^ | j t
� � d | _ | j | � d S)Nr �=r ) r
r ZNodeZsyms�argumentr 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)r r r Zpgen2r r Z
fixer_utilr r r r r
Zcompile_patternr ZBaseFixr r r r r �<module> s (