a
    zen                     @   sV   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   sh  | j }|d  }|jtjkr2d}| || d S t|r^t|rX|jd jd  }q:d|_d|vrt	
|jtd|g}|j|_|S |d  }t|rdd	 |jdd
 D }nd|_|g}d|v rB|d  }	d|	_|}
|jtjks|jdkrt||}
t|
tdt|	gg }t	
|jtdg| }|j|_|S t	j
|jtdt||g|jdS 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.9/lib2to3/fixes/fix_raise.py
<listcomp>D       z&FixRaise.transform.<locals>.<listcomp> tbNonewith_traceback)prefix)symsr   typer   STRINGZcannot_convertr	   Z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&   sB    

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   