o
    Th@                     @   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rd}| || d S t|r2t|r/|jd jd  }t|s!d|_d|v r=|d  }nd }d|v r|d  }t|r_dd	 |jdd
 D }t	||}nC|jtj
tjfv rrd|_t	||g}n0|jtjkr~|jdkr~n$tdd| d|_|t |g}|d ur|t |g7 }t	td||jdS |d urd|_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   [/home/air/segue/gpt/backup/venv/lib/python3.10/site-packages/libfuturize/fixes/fix_raise.py
<listcomp>I   s    z&FixRaise.transform.<locals>.<listcomp> Nonezfuture.utilsraise_)prefixwith_tracebackraise)symsr   typer   STRINGcannot_convertr   childrenr   r   NUMBERNAMEvaluer   r   r   r	   r
   r   Node
raise_stmt)
selfnoderesultsr   r   msgr   r   argsexc_listr   r   r   	transform*   sL   	zFixRaise.transformN)__name__
__module____qualname__BM_compatiblePATTERNr/   r   r   r   r   r   #   s    r   N)__doc__lib2to3r   r   lib2to3.pgen2r   lib2to3.fixer_utilr   r   r   r   r	   r
   libfuturize.fixer_utilr   BaseFixr   r   r   r   r   <module>   s     