U
    ËZj8  ã                   @   s\   d Z ddlmZmZmZ ddlmZ ddlmZm	Z	m
Z
mZ e d¡ZG dd„ dejƒZdS )	a,  Fixer for print.

Change:
    "print"          into "print()"
    "print ..."      into "print(...)"
    "print(...)"     not changed
    "print ... ,"    into "print(..., end=' ')"
    "print >>x, ..." into "print(..., file=x)"

No changes are applied if print_function is imported from __future__

é    )ÚpatcompÚpytreeÚ
fixer_base)Útoken)ÚNameÚCallÚCommaÚStringz8atom< '(' [arith_expr|atom|power|term|STRING|NAME] ')' >c                   @   s$   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”  |st ‚| d¡}|r4| ttdƒg |jd�¡ d S |jd tdƒksJt ‚|jdd … }t|ƒdkrvt 	|d ¡rvd S d  } }}|r¤|d t
ƒ kr¤|d d… }d}|ræ|d t tjd¡kræt|ƒd	ksÎt ‚|d  ¡ }|d
d … }dd„ |D ƒ}|�rd|d _|d k	�s"|d k	�s"|d k	�rz|d k	�rB|  |dtt|ƒƒ¡ |d k	�rb|  |dtt|ƒƒ¡ |d k	�rz|  |d|¡ ttdƒ|ƒ}	|j|	_|	S )NÚbareÚprint)Úprefixr   é   éÿÿÿÿú z>>é   é   c                 S   s   g | ]}|  ¡ ‘qS © )Úclone)Ú.0Úargr   r   ú^/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/libfuturize/fixes/fix_print.pyÚ
<listcomp>A   s     z&FixPrint.transform.<locals>.<listcomp>Ú ÚsepÚendÚfile)ÚAssertionErrorÚgetÚreplacer   r   r   ÚchildrenÚlenÚparend_exprÚmatchr   r   ÚLeafr   Ú
RIGHTSHIFTr   Ú	add_kwargr	   Úrepr)
ÚselfÚnodeÚresultsZ
bare_printÚargsr   r   r   Zl_argsZn_stmtr   r   r   Ú	transform$   s@    
ÿ



zFixPrint.transformc                 C   sN   d|_ t | jjt|ƒt tjd¡|f¡}|r@| 	t
ƒ ¡ d|_ | 	|¡ d S )Nr   ú=r   )r   r   ÚNodeZsymsÚargumentr   r$   r   ÚEQUALÚappendr   )r(   Zl_nodesZs_kwdZn_exprZ
n_argumentr   r   r   r&   T   s    
þÿzFixPrint.add_kwargN)Ú__name__Ú
__module__Ú__qualname__ZBM_compatibleZPATTERNr,   r&   r   r   r   r   r
      s   0r
   N)Ú__doc__Zlib2to3r   r   r   Zlib2to3.pgen2r   Zlib2to3.fixer_utilr   r   r   r	   Úcompile_patternr"   ZBaseFixr
   r   r   r   r   Ú<module>   s   ÿ