B
    18\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   sl  | j }|d  }|jtjkr2d}| || d S t|rbx t|rZ|jd jd  }q<W d|_d|krt	
|jtd|g}|j|_|S |d  }t|rdd	 |jdd
 D }nd|_|g}d|krF|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   [/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/lib2to3/fixes/fix_raise.py
<listcomp>D   s    z&FixRaise.transform.<locals>.<listcomp> tbNonewith_traceback)prefix)symsr   typer   STRINGcannot_convertr	   childrenr   r   Node
raise_stmtr   NAMEvaluer   r   r   simple_stmt)selfnoderesultsr   r   msgnewr   argsr   eZwith_tbr   r   r   	transform&   s@    


zFixRaise.transformN)__name__
__module____qualname__BM_compatiblePATTERNr.   r   r   r   r   r
      s   r
   N)__doc__r   r   pgen2r   r   
fixer_utilr   r   r   r   r	   BaseFixr
   r   r   r   r   <module>   s
   