U

i�fn�@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Csh
|j}|d
�
�}|jtjkr2d}|�|
|�
dSt|�
r^t|�
rX|jdjd�
�}q:d|_d|kr�t	�
|jt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|_|St	j
|jtd�
t||�g|
jd�SdS)N�excz+Python 3 does not support string exceptions�
��
 �val�raisec
Ssg|]}
|
���qS�)
�clone)�.0�
crr�</opt/alt/python38/lib64/python3.8/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&sB






















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