a
    W%b,                     @   sb   d Z ddlmZmZ ddlmZ ddlmZmZm	Z	m
Z
mZmZ ddlmZ G dd dejZdS )	a  Fixer for 'raise E, V'

From Armin Ronacher's ``python-modernize``.

raise         -> raise
raise E       -> raise E
raise E, 5    -> raise E(5)
raise E, 5, T -> raise E(5).with_traceback(T)
raise E, None, T -> raise E.with_traceback(T)

raise (((E, E'), E''), E'''), 5 -> raise E(5)
raise "foo", V, T               -> warns about string exceptions

raise E, (V1, V2) -> raise E(V1, V2)
raise E, (V1, V2), T -> raise E(V1, V2).with_traceback(T)


CAVEATS:
1) "raise E, V, T" cannot be translated safely in general. If V
   is not a tuple or a (number, string, None) literal, then:

   raise E, V, T -> from future.utils import raise_
                    raise_(E, V, T)
    )pytree
fixer_base)token)NameCallis_tupleCommaAttrArgList)touch_import_topc                   @   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   s  | j }|d  }|jtjkr2d}| || d S t|r^t|rX|jd jd  }q:d|_d|v rt|d  }nd }d|v r<|d  }t|rdd	 |jdd
 D }t	||}n|jtj
tjfv rd|_t	||g}n\|jtjkr|jdkrnDtdd| d|_|t |g}|d ur.|t |g7 }t	td|S |d urhd|_t|tdt|gg }	n|g}	tj|jtdg|	 |jdS )Nexcz+Python 3 does not support string exceptions   r    tbvalc                 S   s   g | ]}|  qS  )clone).0cr   r   k/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/libfuturize/fixes/fix_raise.py
<listcomp>I       z&FixRaise.transform.<locals>.<listcomp> Nonezfuture.utilsZraise_with_tracebackraise)prefix)symsr   typer   STRINGZcannot_convertr   childrenr   r   NUMBERNAMEvaluer   r   r   r	   r
   r   NodeZ
raise_stmt)
selfnoderesultsr   r   msgr   r   argsZexc_listr   r   r   	transform*   sJ    
	

zFixRaise.transformN)__name__
__module____qualname__ZBM_compatibleZPATTERNr,   r   r   r   r   r   #   s   r   N)__doc__Zlib2to3r   r   Zlib2to3.pgen2r   Zlib2to3.fixer_utilr   r   r   r   r	   r
   Zlibfuturize.fixer_utilr   ZBaseFixr   r   r   r   r   <module>   s
    