3
\n � @ sZ d Z ddlmZ ddlmZ ddlmZ ddlmZmZm Z m
Z
m Z G dd� dej �Z
dS ) a[ Fixer for 'raise E, V, T'
raise -> raise
raise E -> raise E
raise E, V -> raise E(V)
raise E, V, T -> raise E(V).with_traceback(T)
raise E, None, T -> raise E.with_traceback(T)
raise (((E, E'), E''), E'''), V -> raise E(V)
raise "foo", V, T -> warns about string exceptions
CAVEATS:
1) "raise E, V" will be incorrectly translated if V is an exception
instance. The correct Python 3 idiom is
raise E from V
but since we can't detect instance-hood by syntax alone and since
any client code would have to be changed as well, we don't automate
this.
� )�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� }q<W d|_d|kr�t j
|j t d�|g�}|j|_|S |d j� }t|�r�dd � |jdd� D �}n d
|_|g}d |k�rF|d j� } d
| _|}
|jtj
kp�|jd k�rt||�}
t|
t d
��t| g�g } t j
|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 | ] }|j � �qS � )�clone)�.0�cr r �//usr/lib64/python3.6/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& s@
zFixRaise.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr' r r r r r
s r
N)�__doc__r r Zpgen2r r Z
fixer_utilr r r r r ZBaseFixr
r r r r �<module> s