U
��,an � @ sV d dl mZ d dlmZ d dl mZ d dlmZmZmZm Z m
Z
G dd� dej �Z dS )� )�pytree)�token)�
fixer_base)�Name�Call�Attr�ArgList�is_tuplec @ s e Zd ZdZdZdd� ZdS )�FixRaiseTzB
raise_stmt< 'raise' exc=any [',' val=any [',' tb=any]] >
c C sh | j }|d �� }|jtjkr2d}| �||� d S t|�r^t|�rX|jd jd �� }q:d|_d|kr�t �
|j t d�|g�}|j|_|S |d �� }t|�r�dd � |jdd
� D �}n d |_|g}d |k�rB|d �� } d | _|}
|jtj
ks�|jd
k�rt||�}
t|
t d��t| g�g } t �
|jt d�g| �}|j|_|S t j
|j t d�t||�g|jd�S d S )N�excz+Python 3 does not support string exceptions� � � �val�raisec S s g | ] }|� � �qS � )�clone)�.0�cr r �</opt/alt/python38/lib64/python3.8/lib2to3/fixes/fix_raise.py�
<listcomp>D s z&FixRaise.transform.<locals>.<listcomp>���� �tb�None�with_traceback)�prefix)�symsr �typer �STRINGZcannot_convertr Zchildrenr r ZNodeZ
raise_stmtr �NAME�valuer r r Z simple_stmt) �selfZnodeZresultsr r �msg�newr �argsr �eZwith_tbr r r � transform& sB
�zFixRaise.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr' r r r r r
s r
N)
r r Zpgen2r r Z
fixer_utilr r r r r ZBaseFixr
r r r r �<module> s