a XC?hn�@sVddlmZddlmZddlmZddlmZmZmZm Z m Z Gdd�dej�ZdS)�)�pytree)�token)� fixer_base)�Name�Call�Attr�ArgList�is_tuplec@seZdZdZdZdd�ZdS)�FixRaiseTzB raise_stmt< 'raise' exc=any [',' val=any [',' tb=any]] > cCsh|j}|d��}|jtjkr2d}|�||�dSt|�r^t|�rX|jdjd��}q:d|_d|vr�t � |jtd�|g�}|j|_|S|d��}t|�r�dd �|jdd �D�}nd|_|g}d|v�rB|d��} d| _|} |jtj ks�|jd k�rt||�} t| td��t| g�g}t � |jtd�g|�}|j|_|St j |jtd�t||�g|jd�SdS)N�excz+Python 3 does not support string exceptions�� val�raisecSsg|]}|��qS�)�clone)�.0�crr�D�z&FixRaise.transform..��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 sr N) rrZpgen2rrZ fixer_utilrrrrr ZBaseFixr rrrr�s