B YI~dn�@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]] > cCsl|j}|d��}|jtjkr2d}|�||�dSt|�rbx t|�rZ|jdjd��}qDsz&FixRaise.transform..��tb�None�with_traceback)�prefix)�symsr�typer�STRINGZcannot_convertr ZchildrenrrZNodeZ raise_stmtr�NAME�valuerrrZsimple_stmt)�selfZnode�resultsrr�msg�newr�argsr�eZwith_tbrrr� transform&s@ zFixRaise.transformN)�__name__� __module__�__qualname__Z BM_compatibleZPATTERNr(rrrrr sr N) rrZpgen2rrZ fixer_utilrrrrr ZBaseFixr rrrr�s