3
\n � @ 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 sl | j }|d j� }|jtjkr2d}| j||� d S t|�rbx t|�rZ|jd jd j� }qD s z&FixRaise.transform..� �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& s@
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 � s