�
��Vc� 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 sv � t | t j � � o | j d j t
j k S )N� )�
isinstancer �Node�children�typer ZSTRING)�stmt� �7/usr/lib64/python3.11/lib2to3/fixes/fix_tuple_params.py� is_docstringr s/ � � �d�F�K� (� (� 1� �=�� � �E�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� � ||� � 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 )NZlambda�suite�argsr
� r z; � Fc s� �� t �� � � � � }| � � � }d|_ t ||� � � � � }|rd|_ | � |� � �� t j t 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�y�)�)�D��
� ��� � � �a� � � � � � �V�[��)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. sZ ���� � �w� � 8��(�(��w�7�7� 7�� ��� ���v��� ��8� �Q� � $�� � 4� 0��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 s � |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 s6 � g | ]}|� � � ��S � )r �Z.0Zc� 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 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 sf � ��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)Z__name__Z
__module__Z __qualname__Z run_orderZ
BM_compatibleZPATTERNr, r& r/ r r r r sD � � � � � ��I��M��G�>� >� >�@ � � � � r r c s � | 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 Zvfplistr r4 �vfpdefr
Z RuntimeError�r) r r r3 r3 � s{ � � �y�T�\�5�:�.�.� �� �
��d�k� !� � �i�4�;�&� $��=��#�D� �i�4�;�&� $�� �
�4�t�;�
<�
<�<r c s� � | j t j k rt | j d � � S | j t
j k r| j S d� | j D � � S )Nr c sV � g | ]&}|j t j k �t |� � ��'S r/ )r r ZCOMMAr5 r0 r1 r r2 z find_params.<locals>.<listcomp>� s, � � K� K� K�q�Q�V�u�{�5J� K�K��N�N� K� K� Kr )r r r8 r5 r
r r4 r' r9 r r r5 r5 � sW � � �y�D�K� � ��4�=��+�,�,�,�
��e�j� � ��z�� K� K�D�M� K� K� K�Kr Nc s� � |�i }t | � � D ]_\ }}t t t |� � � � � � g}t |t
� � rt
|||�� � �W||z ||< �`|S )N)�d)r( r r Zstrr �listr6 )�
param_listr r: r+ �objZtrailers r r6 r6 � s� � �� � ���J�'�'� &� &���3� �V�C��F�F�^�^�,�,�-�� �c�4� � � &� ��g�� +� +� +� +� +��g�%�A�c�F�F� �Hr c s� � g }| D ]O}t |t � � r#|� t |� � � � �:|� |� � �Pd� |� � S )NZ_)r r; r r7 Zjoin)r< Zlr= s r r7 r7 � sd � �
�A�� � �� �c�4� � � �
�H�H�Z��_�_� %� %� %� %�
�H�H�S�M�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 � g �X =� =� =�L� L� L� %'�$�
�
�
�
�� � � � r