← Back
Editing: fix_tuple_params.cpython-311.pyc
� �0�i� � � � 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 �v � t | t j � � o | j d j t j k S )N� )� isinstancer �Node�children�typer �STRING)�stmts �E/opt/alt/python311/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 )�__doc__r r �pgen2r r � fixer_utilr r r r r r r �BaseFixr rL rN rO rP rG r r �<module>rv 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
Save File
Cancel