U
    A`n                     @   sZ   d Z 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 )	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                   @   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|krt	
|jtd|g}|j|_|S |d  }t|rdd	 |jdd
 D }nd|_|g}d|k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.8/lib2to3/fixes/fix_raise.py
<listcomp>D   s     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nodeZ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)__doc__r   r   Zpgen2r   r   Z
fixer_utilr   r   r   r   r	   ZBaseFixr
   r   r   r   r   <module>   s
   