�
� f� � � � d Z ddlmZ ddlmZ ddlmZ ddlmZmZmZ ddlmZ ej dhz Z
G d� d ej � � Z d
S ) aj Fixer for dict methods.
d.keys() -> list(d.keys())
d.items() -> list(d.items())
d.values() -> list(d.values())
d.iterkeys() -> iter(d.keys())
d.iteritems() -> iter(d.items())
d.itervalues() -> iter(d.values())
d.viewkeys() -> d.keys()
d.viewitems() -> d.items()
d.viewvalues() -> d.values()
Except in certain very specific contexts: the iter() can be dropped
when the context is list(), sorted(), iter() or for...in; the list()
can be dropped when the context is list() or sorted() (but not iter()
or for...in!). Special contexts that apply to both: list(), sorted(), tuple()
set(), any(), all(), sum().
Note: iter(d.keys()) could be written as iter(d) but since the
original d.iterkeys() was also redundant we don't fix this. And there
are (rare) contexts where it makes a difference (e.g. when passing it
as an argument to a function that introspects the argument).
� )�pytree)�patcomp)�
fixer_base)�Name�Call�Dot)�
fixer_util�iterc �j � e Zd ZdZdZd� ZdZ ej e� � Z dZ
ej e
� � Z d� Z dS )�FixDictTa
power< head=any+
trailer< '.' method=('keys'|'items'|'values'|
'iterkeys'|'iteritems'|'itervalues'|
'viewkeys'|'viewitems'|'viewvalues') >
parens=trailer< '(' ')' >
tail=any*
>
c � � |d }|d d }|d }| j }|j }|� d� � }|� d� � } |s| r
|dd � }|dv sJ t |� � � � �d � |D � � }d
� |D � � }| o| � ||� � }
|t j |j t � � t ||j
� � � g� � |d � � � gz } t j |j | � � } |
s+| s)d
| _
t t |rdnd� � | g� � } |r t j |j | g|z � � } |j
| _
| S )N�head�method� �tailr
�view� )�keys�items�valuesc �6 � g | ]}|� � � ��S � ��clone��.0�ns �=/opt/alt/python311/lib64/python3.11/lib2to3/fixes/fix_dict.py�
<listcomp>z%FixDict.transform.<locals>.<listcomp>A � � �(�(�(�a���� � �(�(�(� c �6 � g | ]}|� � � ��S r r r s r r z%FixDict.transform.<locals>.<listcomp>B r r! )�prefix�parens� �list)�syms�value�
startswith�repr�in_special_contextr �Node�trailerr r r# r �powerr )
�self�node�resultsr r r r' � method_name�isiter�isview�special�args�news
r � transformzFixDict.transform6 s� � ��v�����"�1�%���v����y�� �l� � �'�'��/�/�� �'�'��/�/�� � *�V� *�%�a�b�b�/�K�� 9�9�9�9�4��<�<�9�9�9�(�(�4�(�(�(��(�(�4�(�(�(���(�D�t� 6� 6�t�V� D� D���v�{�4�<�$'�E�E�$(��06�
�%?� %?� %?�$@�A� A� �x�(�.�.�0�0� 2� 2��
�k�$�*�d�+�+��� B�6� B��C�J��t�f� 8�F�F�&�9�9�C�5�A�A�C� � 8��+�d�j�3�%�$�,�7�7�C��[��
��
r! z3power< func=NAME trailer< '(' node=any ')' > any* >zmfor_stmt< 'for' any 'in' node=any ':' any* >
| comp_for< 'for' any 'in' node=any any* >
c �H � |j �dS i }|j j �^| j � |j j |� � r9|d |u r/|r|d j t v S |d j t
j v S |sdS | j � |j |� � o |d |u S )NFr0 �func)�parent�p1�matchr( � iter_exemptr �consuming_calls�p2)r/ r0 r3 r1 s r r+ zFixDict.in_special_contextZ s� � � �;� ��5��� �K� � *��w�}�}�T�[� /��9�9�
+��v��$�&�&��
K� �v��,� �;�;� �v��,�
�0J�J�J�� ��5��w�}�}�T�[�'�2�2�N�w�v��$�7N�Nr! N)
�__name__�
__module__� __qualname__�
BM_compatible�PATTERNr8 �P1r �compile_patternr< �P2r@ r+ r r! r r r ) sw � � � � � ��M��G�� � �8
?�B� �� �� $� $�B�
�B�
!�� �� $� $�B�O� O� O� O� Or! r N)
�__doc__r% r r r r r r r r? r> �BaseFixr r r! r �<module>rK s� ��� �6 � � � � � � � � � � � � � � � � � � (� (� (� (� (� (� (� (� (� (� � � � � � � �(�F�8�3� �AO� AO� AO� AO� AO�j� � AO� AO� AO� AO� AOr!