U
e5d
� @ sb d dl mZ d dlmZ d dl mZ d dlmZmZmZm Z m
Z
m Z dd� Z G dd� dej
�Zd S )
� )�pytree)�token)�
fixer_base)�Assign�Attr�Name�is_tuple�is_list�symsc c sD t | �D ]6\}}|jtjkr|jd jdkr|| |d fV qd S )N� �exceptr )� enumerate�typer
�
except_clause�children�value)Znodes�i�n� r �0/usr/lib64/python3.8/lib2to3/fixes/fix_except.py� find_excepts s r c @ s e Zd ZdZdZdd� ZdS )� FixExceptTa1
try_stmt< 'try' ':' (simple_stmt | suite)
cleanup=(except_clause ':' (simple_stmt | suite))+
tail=(['except' ':' (simple_stmt | suite)]
['else' ':' (simple_stmt | suite)]
['finally' ':' (simple_stmt | suite)]) >
c C sx | j }dd� |d D �}dd� |d D �}t|�D �]\}}t|j�dkr2|jdd� \}} }
| �tdd d
�� |
jtjk�r8t| � � d d
�} |
�
� } d | _ |
�| � | �
� } |j}
t |
�D ]\}}t
|tj�r� q�q�t|
�s�t|
�r�t| t| td ���}n
t| | �}t|
d |� �D ]}|�d
|� �q|�||� q2|
j d kr2d |
_ q2dd� |jd d� D �| | }t�|j|�S )Nc S s g | ] }|� � �qS r ��clone)�.0r r r r �
<listcomp>2 s z'FixExcept.transform.<locals>.<listcomp>�tailc S s g | ] }|� � �qS r r )r Zchr r r r 4 s Zcleanup� � �as� )�prefix� �argsr c S s g | ] }|� � �qS r r )r �cr r r r \ s � )r
r �lenr �replacer r r �NAME�new_namer r! r
�
isinstancer ZNoder r r r �reversedZ insert_child)�selfZnodeZresultsr
r Z try_cleanupr Ze_suite�EZcomma�NZnew_N�targetZ suite_stmtsr ZstmtZassignZchildr r r r � transform/ s6
zFixExcept.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr0 r r r r r $ s r N)r" r Zpgen2r r Z
fixer_utilr r r r r r
r ZBaseFixr r r r r �<module> s