o
     JAfn                     @   sX   	 d dl mZ d dlmZ d dl mZ d dlmZmZmZm	Z	m
Z
 G dd dejZdS )   )pytree)token)
fixer_base)NameCallAttrArgListis_tuplec                   @   s   e Zd ZdZdZdd ZdS )FixRaiseTzB
    raise_stmt< 'raise' exc=any [',' val=any [',' tb=any]] >
    c                 C   sf  | j }|d  }|jtjkrd}| || d S t|r2t|r/|jd jd  }t|s!d|_d|vrGt	
|jtd|g}|j|_|S |d  }t|r^dd	 |jdd
 D }nd|_|g}d|v r|d  }	d|	_|}
|jtjks~|jdkrt||}
t|
tdt|	gg }t	
|jtdg| }|j|_|S t	j
|jtdt||g|jdS )Nexcz+Python 3 does not support string exceptions        valraisec                 S   s   g | ]}|  qS  )clone).0cr   r   ./usr/lib/python3.10/lib2to3/fixes/fix_raise.py
<listcomp>D   s    z&FixRaise.transform.<locals>.<listcomp> tbNonewith_traceback)prefix)symsr   typer   STRINGZcannot_convertr	   childrenr   r   ZNodeZ
raise_stmtr   NAMEvaluer   r   r   Zsimple_stmt)selfZnoderesultsr   r   msgnewr   argsr   eZwith_tbr   r   r   	transform&   sD   
zFixRaise.transformN)__name__
__module____qualname__ZBM_compatibleZPATTERNr)   r   r   r   r   r
      s    r
   N)r   r   Zpgen2r   r   Z
fixer_utilr   r   r   r   r	   ZBaseFixr
   r   r   r   r   <module>   s    