�
� f�
� �Z � d dl mZ d dlmZmZmZ d dl mZ G d� dej � � ZdS )� )�
fixer_base)�Name�Call�consuming_calls)�patcompc � � � e Zd ZdZdZ� fd�Zd� Zd� Zd� Zd� Z dZ
e j e
� � Z
d Z e j e� � Zd
� Z� xZS ) � FixXrangeTz�
power<
(name='range'|name='xrange') trailer< '(' args=any ')' >
rest=any* >
c � �� t t | � � � ||� � t � � | _ d S �N)�superr �
start_tree�set�transformed_xranges)�self�tree�filename� __class__s ��?/opt/alt/python311/lib64/python3.11/lib2to3/fixes/fix_xrange.pyr
zFixXrange.start_tree s5 �� �
�i�� � �)�)�$��9�9�9�#&�5�5�� � � � c � � d | _ d S r )r )r r r s r � finish_treezFixXrange.finish_tree s � �#'�� � � r c �� � |d }|j dk r| � ||� � S |j dk r| � ||� � S t t |� � � � �)N�name�xrange�range)�value�transform_xrange�transform_range�
ValueError�repr�r �node�resultsr s r � transformzFixXrange.transform se � ��v��� �:�� !� !��(�(��w�7�7� 7�
�Z�7�
"�
"��'�'��g�6�6� 6� �T�$�Z�Z�(�(� (r c � � |d }|� t d|j �� � � � | j � t |� � � � d S )Nr r ��prefix)�replacer r' r �add�idr! s r r zFixXrange.transform_xrange$ sO � ��v��� � � �T�'�$�+�6�6�6�7�7�7� � �$�$�R��X�X�.�.�.�.�.r c �Z � t |� � | j vr�| � |� � st t d� � |d � � � g� � }t t d� � |g|j �� � }|d D ]}|� |� � �|S d S d S )Nr �args�listr& �rest)r* r �in_special_contextr r �cloner' � append_child)r r" r# �
range_call� list_call�ns r r zFixXrange.transform_range* s� � � �t�H�H�D� 4� 4� 4��'�'��-�-�
5� �d�7�m�m�g�f�o�.C�.C�.E�.E�-F�G�G�J� �T�&�\�\�J�<�$(�K�1� 1� 1�I� �V�_�
*�
*���&�&�q�)�)�)�)� � �
5� 4� 4� 4r z3power< func=NAME trailer< '(' node=any ')' > any* >z�for_stmt< 'for' any 'in' node=any ':' any* >
| comp_for< 'for' any 'in' node=any any* >
| comparison< any 'in' node=any any*>
c �
� |j �dS i }|j j �C| j � |j j |� � r |d |u r|d j t v S | j � |j |� � o |d |u S )NFr" �func)�parent�p1�matchr r �p2)r r" r# s r r/ z FixXrange.in_special_context? s� � � �;� ��5��� �K� � *��w�}�}�T�[� /��9�9�
+��v��$�&�&��6�?�(�O�;� ;��w�}�}�T�[�'�2�2�N�w�v��$�7N�Nr )�__name__�
__module__� __qualname__�
BM_compatible�PATTERNr
r r$ r r �P1r �compile_patternr8 �P2r: r/ �
__classcell__)r s @r r r s� �� � � � � ��M��G� )� )� )� )� )�(� (� (�)� )� )�/� /� /�
�
�
�
?�B� �� �� $� $�B�
�B�
!�� �� $� $�B�
O�
O�
O�
O�
O�
O�
Or r N) � r �
fixer_utilr r r r �BaseFixr � r r �<module>rH s� �� 6� � � � � � � 4� 4� 4� 4� 4� 4� 4� 4� 4� 4� � � � � � �=O� =O� =O� =O� =O�
�"� =O� =O� =O� =O� =Or