�
� f� � � � d dl mZ d dlmZ d dl mZ d dlmZmZmZm Z m
Z
m Z d� Z G d� dej
� � Zd� Zd � Zg d
fd �Zd � Zd
S )
� )�pytree)�token)�
fixer_base)�Assign�Name�Newline�Number� Subscript�symsc �v � t | t j � � o | j d j t
j k S )N� )�
isinstancer �Node�children�typer �STRING)�stmts �7/usr/lib64/python3.11/lib2to3/fixes/fix_tuple_params.py� is_docstringr s/ � � �d�F�K� (� (� 1� �=�� � �E�L� 0�1� c �& � e Zd ZdZdZdZd� Zd� ZdS )�FixTupleParams� Ta
funcdef< 'def' any parameters< '(' args=any ')' >
['->' any] ':' suite=any+ >
|
lambda=
lambdef< 'lambda' args=vfpdef< '(' inner=any ')' >
':' body=any
>
c � � � �
� d|v r� � ||� � S g �
|d }|d }|d j d j t j k r)d}|d j d j }t
� � � n#d}d}t j t j d� � � d
� �
� fd
� }|j t j
k r ||� � nU|j t j k r@t |j � � D ]+\ }} | j t j
k r || |dk � � � �,�
sd S �
D ]}
|d |
_
�|} |dk rd �
d _ n2t |d j | � � r|�
d _ |dz } �
D ]}
|d |
_
��
|d j | | �< t! | dz | t# �
� � z dz � � D ]}||d j | _ �|d � � � d S )N�lambda�suite�argsr
� r z; � Fc � �� t �� � � � � }| � � � }d|_ t ||� � � � � }|rd|_ | � |� � �� t j t j
|�� � � g� � � � d S )Nr � ) r �new_name�clone�prefixr �replace�appendr r r � simple_stmt)� tuple_arg�
add_prefix�n�argr �end� new_lines�selfs ���r � handle_tuplez.FixTupleParams.transform.<locals>.handle_tupleC s� �� ��T�]�]�_�_�%�%�A��/�/�#�#�C��C�J��#�q�w�w�y�y�)�)�D��
� ��� � � �a� � � � � � �V�[��)9�*.�� � � � �)<� >� >�
?�
?�
?�
?�
?r )r) r! )F)�transform_lambdar r r �INDENT�valuer r �Leafr �tfpdef�
typedargslist� enumerate�parentr$ r �range�len�changed)r. �node�resultsr r �start�indentr/ �ir+ �line�afterr, r- s ` @@r � transformzFixTupleParams.transform. sM ���� � �w� � ��(�(��w�7�7� 7�� ��� ���v��� ��8� �Q� � $�� � 4� 4��E��1�X�&�q�)�/�F��)�)�C�C��E��F��+�e�l�B�/�/�C�
?� ?� ?� ?� ?� ?� ?� ?� �9�� � #� #� �L�� � � � �
�Y�$�,�
,�
,�#�D�M�2�2�
:�
:���3��8�t�{�*�*� !�L��!�a�%�9�9�9�9��� � �F� � #� #�D� ��(�D�K�K� �� �A�:�:�"%�I�a�L� � �
�%��(�+�E�2�
3�
3� �"(�I�a�L� ��A�I�E� � #� #�D� ��(�D�K�K�)2��a���%��+�&��u�Q�w��c�)�n�n� 4�Q� 6�7�7� 1� 1�A�*0�E�!�H� �a� � '� '�
�a��������r c � � |d }|d }t |d � � }|j t j k r2|� � � }d|_ |� |� � d S t |� � }t |� � }| � t |� � � � }t |d�� � } |� | � � � � � |� � � D ]�}
|
j t j k rv|
j
|v rmd� ||
j
D � � } t j t j | � � � g| z � � } |
j | _ |
� | � � ��d S )Nr �body�innerr! )r$ c �6 � g | ]}|� � � ��S � )r# ��.0�cs r �
<listcomp>z3FixTupleParams.transform_lambda.<locals>.<listcomp>� s � � C� C� C�A�a�g�g�i�i� C� C� Cr )�
simplify_argsr r �NAMEr# r$ r% � find_params� map_to_indexr" �
tuple_namer �
post_orderr2 r r r �power)
r. r; r< r rD rE �params�to_index�tup_name� new_paramr* �
subscripts�news
r r0 z FixTupleParams.transform_lambdan s` � ��v����v��� �g�g� .�/�/�� �:��� #� #��K�K�M�M�E� �E�L� �L�L�� � � � �F� �T�"�"�� ��'�'���=�=��F�!3�!3�4�4����#�.�.�.� � � � �Y�_�_�&�&�'�'�'�� � �"�"� � �A��v���#�#���8�(;�(;� C� C��!�'�1B� C� C� C�
� �k�$�*�#,�?�?�#4�#4�"5�
�"B�D� D�� �X��
�� � �#�����
� r N)�__name__�
__module__� __qualname__� run_order�
BM_compatible�PATTERNrB r0 rG r r r r sD � � � � � ��I��M��G�>� >� >�@ � � � � r r c � � | j t j t j fv r| S | j t j k r9| j t j k r"| j d } | j t j k �"| S t d| z � � �)Nr zReceived unexpected node %s)r r �vfplistr rM �vfpdefr � RuntimeError�r; s r rL rL � ss � � �y�T�\�5�:�.�.�.�� �
��d�k� !� !� �i�4�;�&�&��=��#�D� �i�4�;�&�&�� �
�4�t�;�
<�
<�<r c � � | j t j k rt | j d � � S | j t
j k r| j S d� | j D � � S )Nr c �V � g | ]&}|j t j k �t |� � ��'S rG )r r �COMMArN rH s r rK z find_params.<locals>.<listcomp>� s, � � K� K� K�q�Q�V�u�{�5J�5J�K��N�N�5J�5J�5Jr )r r ra rN r r rM r2 rc s r rN rN � sS � � �y�D�K� � ��4�=��+�,�,�,�
��e�j� � ��z�� K� K�D�M� K� K� K�Kr Nc �� � |�i }t | � � D ]_\ }}t t t |� � � � � � g}t |t
� � rt
|||�� � �W||z ||< �`|S )N)�d)r6 r
r �strr �listrO )�
param_listr$ rh r? �obj�trailers r rO rO � s� � ��y� ���J�'�'� &� &���3� �V�C��F�F�^�^�,�,�-�� �c�4� � � &� ��g�� +� +� +� +� +��g�%�A�c�F�F� �Hr c �� � g }| D ]O}t |t � � r#|� t |� � � � �:|� |� � �Pd� |� � S )N�_)r rj r& rP �join)rk �lrl s r rP rP � sd � �
�A�� � �� �c�4� � � �
�H�H�Z��_�_� %� %� %� %�
�H�H�S�M�M�M�M� �8�8�A�;�;�r )r r �pgen2r r �
fixer_utilr r r r r
r r �BaseFixr rL rN rO rP rG r r �<module>ru s ���* � � � � � � � � � � � � � � � � � � G� G� G� G� G� G� G� G� G� G� G� G� G� G� G� G�1� 1� 1�g � g � g � g � g �Z�'� g � g � g �X =� =� =�L� L� L� %'�$�
�
�
�
�� � � � r