�
� Wc� c s� � d Z ddlmZ ddlmZ ddlmZ ddlmZmZm Z m
Z
m Z m Z d� Z
G d� dej � Zd � Zd
� Zg d fd �Zd
� Zd S )a: Fixer for function definitions with tuple parameters.
def func(((a, b), c), d):
...
->
def func(x, d):
((a, b), c) = x
...
It will also support lambdas:
lambda (x, y): x + y -> lambda t: t[0] + t[1]
# The parens are a syntax error in Python 3
lambda (x): x + y -> lambda x: x + y
� )�pytree)�token)�
fixer_base)�Assign�Name�Newline�Number� Subscript�symsc s� � t | t j � o.| j d j t
j k S )N� )�
isinstancer �Node�children�typer ZSTRING)�stmt� �7/usr/lib64/python3.12/lib2to3/fixes/fix_tuple_params.py� is_docstringr s5 � � �d�F�K�K� (� 1� �=�=�� � � �E�L�L� 0�1� c s& � e Zd ZdZdZdZd� Zd� ZdS )�FixTupleParamsi Ta
funcdef< 'def' any parameters< '(' args=any ')' >
['->' any] ':' suite=any+ >
|
lambda=
lambdef< 'lambda' args=vfpdef< '(' inner=any ')' >
':' body=any
>
c s� � � �
� d|v r� j ||� S g �
|d }|d }|d j d j t j k r1d}|d j d j
}t
� � n+d}d}t j t j d� � d
� �
� fd
� }|j t j k r
||� nl|j t j k rMt |j � D ]4 \ }} | j t j k s�& || |dk � � �6�
sd S �
D ] }
|d |
_
�|} |dk rd �
d _ n5t |d j | � r|�
d _ |dz } �
D ] }
|d |
_
��
|d j | | t! | dz | t# �
� z dz � D ] }||d j | _ �!|d j% � d S )NZlambda�suite�argsr
� r z; � Fc sp �� t �j � � }| j � }d|_ t ||j � � }|rd|_ | j |� �j
t j t j |�j � g� � d S )Nr � ) r �new_name�clone�prefixr �replace�appendr r r Z simple_stmt)Z 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�)�D��z� ��� � � �a� � � � � �V�[�[��)9�)9�*.�� � � �)<� >�
?�
?�
?�
?r )r r )F)�transform_lambdar
r r ZINDENT�valuer r ZLeafr ZtfpdefZ
typedargslist� enumerateZparentr r ZrangeZlenZchanged)r$ �node�resultsr r ZstartZindentr% �ir! ZlineZafterr" r# s ` @@r � transformzFixTupleParams.transform. sW ���� � �w� � ��(�(��w�7� 7�� ��� ���v��� ��8� � �Q� � $� $�� � � 4� 4��E��1�X�&�&�q�)�/�/�F��)�C�C��E��F��+�+�e�l�l�B�/�C�
?� ?� ?� ?� ?� ?� ?� ?� �9�9�� � � #� #� �L�� � � �
�Y�Y�$�,�,�
,�
,�#�D�M�M�2�
:�
:���3��8�8�t�{�{�*�*�� !�L��!�a�%�9�9�9�9��y� �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� 4�Q� 6�7� 1� 1�A�*0�E�!�H� � �a� � '� '�
�a�������r c s& � |d }|d }t |d � }|j t j k r,|j � }d|_ |j
|� d S t |� }t |� }| j t |� � }t |d�� } |j
| j � � |j � D ]� }
|
j t j k s�#|
j |v s�2d� ||
j D � } t j t j" | j � g| z � } |
j
| _ |
j
| � ��d S )Nr �body�innerr )r c s2 � g | ] }|j � ��S � )r �Z.0Zc� r �
<listcomp>z3FixTupleParams.transform_lambda.<locals>.<listcomp>� s � � C� C� C�A�a�g�g�i�i�i� Cr )�
simplify_argsr r �NAMEr r r � find_params� map_to_indexr �
tuple_namer Z
post_orderr' r r r Zpower)
r$ r) r* r r- r. ZparamsZto_indexZtup_nameZ new_paramr Z
subscriptsZnews
r r& z FixTupleParams.transform_lambdan sG � ��v����v��� �g�g� .�/�� �:�:���� #� #��K�K�M�E� �E�L� �L�L�� � � �F� �T�"�� ��'���=�=��F�!3�4����#�.�.� � � � �Y�_�_�&�'�'�� � �"� � �A��v�v����#�#�����8�(;�(;�� C� C��!�'�'�1B� C� C�
� �k�k�$�*�*�#,�?�?�#4�"5�
�"B�D�� �X�X��
�� � �#����
� r N)Z__name__Z
__module__Z __qualname__Z run_orderZ
BM_compatibleZPATTERNr, r& r/ r r r r sD � � � � � ��I��M��G�>� >� >�@ � � � � r r c sf � | j t j t j fv r| S | j t j
k rS| j t j
k r2| j d } | j t j
k r�2| S t d| z � �)Nr zReceived unexpected node %s)r r Zvfplistr r4 �vfpdefr
Z RuntimeError�r) r r r3 r3 � s� � � �y�y�T�\�\�5�:�:�.�.�.�� �
���d�k�k� !� !� �i�i�4�;�;�&�&��=�=��#�D� �i�i�4�;�;�&�&��� �
�4�t�;�
<�<r c s� � | j t j k r t | j d � S | j t
j k r | j S d� | j D � S )Nr c sj � g | ]/ }|j t j k s�#t |� ��1S r/ )r r ZCOMMAr5 r0 r1 r r2 z find_params.<locals>.<listcomp>� s- � � K� K� K�q�Q�V�V�u�{�{�5J�5J��K��N�N�N� Kr )r r r8 r5 r
r r4 r' r9 r r r5 r5 � s[ � � �y�y�D�K�K� � ��4�=�=��+�,�,�
���e�j�j� � ��z�z�� K� K�D�M�M� K� K�Kr Nc s� � |�i }t | � D ]U \ }}t t t |� � � g}t |t
� rt
|||�� �N||z ||< �W|S )N)�d)r( r r Zstrr �listr6 )�
param_listr r: r+ �objZtrailers r r6 r6 � sp � ��y� ���J�'� &� &���3� �V�C��F�^�,�-�� �c�4� � � ��g�� +� +� +� +��g�%�A�c�F�F� �Hr c s� � g }| D ]E }t |t � r |j t |� � �4|j |� �Gdj |� S )NZ_)r r; r r7 Zjoin)r< Zlr= s r r7 r7 � sV � �
�A�� � �� �c�4� � �
�H�H�Z��_� %� %� %�
�H�H�S�M�M�M� �8�8�A�;�r )Z__doc__r r Zpgen2r r Z
fixer_utilr r r r r r r ZBaseFixr r3 r5 r6 r7 r/ r r �<module>r> 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 �X =� =� =�L� L� L� %'�$�
�
�
�
�� � � � r