�
� f � � � d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZm Z m
Z
m Z ej d� � Z
G d� 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 �" � e Zd ZdZdZd� Zd� ZdS )�FixPrintTzP
simple_stmt< any* bare='print' any* > | print_stmt
c � � |� d� � }|r9|� t t d� � g |j �� � � � d S |j dd � }t
|� � dk r"t � |d � � rd S d x}x}}|r$|d t � � k r |d d� }d}|rM|d t j t j
d� � k r$|d � � � }|d d � }d
� |D � � }|r
d |d _ |�|�|�|�1| � |d t! t# |� � � � � � |�1| � |d
t! t# |� � � � � � |�| � |d|� � t t d� � |� � } |j | _ | S )N�bare�print)�prefix� � ���� z>>� c �6 � g | ]}|� � � ��S � )�clone)�.0�args �0/usr/lib64/python3.11/lib2to3/fixes/fix_print.py�
<listcomp>z&FixPrint.transform.<locals>.<listcomp>? s � �.�.�.�#�#�)�)�+�+�.�.�.� � �sep�end�file)�get�replacer r r �children�len� parend_expr�matchr r �Leafr �
RIGHTSHIFTr � add_kwargr
�repr)
�self�node�results�
bare_print�argsr r r! �l_args�n_stmts
r � transformzFixPrint.transform% s� � � �[�[��(�(�
� � � � � �t�D��M�M�2�&0�&7� 9� 9� 9�
:�
:�
:� �F��}�Q�R�R� �� �t�9�9��>�>�k� /� /��Q�� 8� 8�>�
�F� � �� �c�D� � �D��H����'�'�����9�D��C� � �D��G�v�{�5�+;�T� B� B�B�B���7�=�=�?�?�D�����8�D�.�.��.�.�.�� � "� !�F�1�I� � �?�c�o��1A������v�u�f�T�#�Y�Y�.?�.?�@�@�@������v�u�f�T�#�Y�Y�.?�.?�@�@�@�� ����v�v�t�4�4�4��d�7�m�m�V�,�,�� � ��
��
r c �* � d|_ t j | j j t |� � t j t j d� � |f� � }|r(|� t � � � � d|_ |� |� � d S )Nr �=r ) r r �Node�syms�argumentr r( r �EQUAL�appendr )r, �l_nodes�s_kwd�n_expr�
n_arguments r r* zFixPrint.add_kwargM s� � ���
��[���!3�"&�u�+�+�"(�+�e�k�3�"?�"?�"(�"*�+� +�
� � $� �N�N�5�7�7� #� #� #� #�J� ����z�"�"�"�"�"r N)�__name__�
__module__� __qualname__�
BM_compatible�PATTERNr3 r* r r r r r s? � � � � � ��M��G�&� &� &�P
#�
#�
#�
#�
#r r N)�__doc__r r r �pgen2r r �
fixer_utilr r r r
�compile_patternr& �BaseFixr r r r �<module>rI s� ��
�
� � � � � � � � � � � � � � � � � � � � � � � � � 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� &�g�%�6�� � �
:#� :#� :#� :#� :#�z�!� :#� :#� :#� :#� :#r