3


 \n�@sZdZd
dl
mZ
d
dlmZ
d
dl
mZ
d
dlmZmZm	Z	m
Z
mZ
Gdd�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@seZ
dZd
ZdZdd�ZdS)�FixRaiseTzB
    raise_stmt< 'raise' exc=any [',' val=any [',' tb=any]] >
    cCsl
|j}|d
j
�}|jtjkr2d}|j|
|�
dSt|�
rbx t|�
rZ|jdjdj
�}q<Wd|_d|kr�t	j
|jtd�
|g�}|
j|_|S|dj
�}t|�
r�dd	�|jdd�D�
}nd
|_|g
}d|k�
rF|dj
�}	d
|	_|}
|jtj
kp�|jdk�
rt||�}
t|
td
�
�t|	g
�
g
}t	j
|jtd�
g
|�}|
j|_|St	j
|jtd�
t||�g|
jd�SdS)N�excz+Python 3 does not support string exceptions�
��
 �val�raisec
Ssg|]}
|
j��qS�)
�clone)�.0�
crr�//usr/lib64/python3.6/lib2to3/fixes/fix_raise.py�
<listcomp>Dsz&FixRaise.transform.<locals>.<listcomp>��tb�None�with_traceback)
�prefix���)�symsr�typer�STRINGZcannot_convertr	ZchildrenrrZNodeZ
raise_stmtr�NAME�valuerrrZsimple_stmt)�selfZnodeZresultsrr�msg�newr�argsr�
eZwith_tbrrr�	transform&s@























zFixRaise.transformN)�__name__�
__module__�__qualname__Z
BM_compatibleZPATTERNr'rrrrr
sr
N)�__doc__rrZpgen2rrZ
fixer_utilrrrrr	ZBaseFixr
rrrr�<module>s