a ij@s|dZddlmZddlZddlmZmZmZmZm Z e e ege ffZ ddddddd d Z ej dddd d d dZdS)zUtils to reraise.) annotationsN)CallableIteratorNoReturnOptionalUnion ExceptionzOptional[_Str]r)eprefixsuffixreturncst|r|n|}t|r |n|}|p*d}|r8d|nd}||}Gfdddt}tj|_tj|_tj|_||}|jjdS)aReraise an exception with an additional message. Benefit: Contrary to `raise ... from ...` and `raise Exception().with_traceback(tb)`, this function will: * Keep the original exception type, attributes,... * Avoid multi-nested `During handling of the above exception, another exception occurred`. Only the single original stacktrace is displayed. This result in cleaner and more compact error messages. Usage: ``` try: fn(x) except Exception as e: epy.reraise(e, prefix=f'Error for {x}: ') ``` Args: e: Exception to reraise prefix: Prefix to add to the exception message. suffix: Suffix to add to the exception message.  cs6eZdZdZddZddfdd ZejZejZdS) z!reraise..WrappedExceptionz(Exception proxy with additional message.cSst||dSN)r__init__)selfmsgrU/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/epy/reraise_utils.pyrQsz*reraise..WrappedException.__init__str)namecs t|Sr)getattr)rrr rr __getattr__Vsz-reraise..WrappedException.__getattr__N) __name__ __module__ __qualname____doc__rr BaseException__repr____str__rrrrWrappedExceptionNs r!N)callabletyperrrwith_traceback __traceback__ __cause__)r r r rr!Z new_exceptionrrrreraise&s    r'zIterator[None])r r r c csBz dVWn2ty<}zt|||dWYd}~n d}~00dS)aContext manager which reraise exceptions with an additional message. Benefit: Contrary to `raise ... from ...` and `raise Exception().with_traceback(tb)`, this function will: * Keep the original exception type, attributes,... * Avoid multi-nested `During handling of the above exception, another exception occurred`. Only the single original stacktrace is displayed. This result in cleaner and more compact error messages. Usage: ```python with epy.maybe_reraise(prefix=f'Error for {x}:'): fn(x) ``` Args: prefix: Prefix to add to the exception message. Can be a function for lazy-evaluation. suffix: Suffix to add to the exception message. Can be a function for lazy-evaluation. Yields: None N)r r )rr')r r r rrr maybe_reraiseos  r()NN)NN)r __future__r contextlibtypingrrrrrrZ_Strr'contextmanagerr(rrrrs I