import sys
from functools import wraps
from . import Traceback
def reraise(tp, value, tb=None):
try:
if value is None:
value = tp()
if value.__traceback__ is not tb:
raise value.with_traceback(tb)
raise value
finally:
value = None
tb = None
class Error:
def __init__(self, exc_type, exc_value, traceback):
self.exc_type = exc_type
self.exc_value = exc_value
self.__traceback = Traceback(traceback)
@property
def traceback(self):
return self.__traceback.as_traceback()
def reraise(self):
reraise(self.exc_type, self.exc_value, self.traceback)
def return_error(func, exc_type=Exception):
@wraps(func)
def return_exceptions_wrapper(*args, **kwargs):
try:
return func(*args, **kwargs)
except exc_type:
return Error(*sys.exc_info())
return return_exceptions_wrapper
returns_error = return_errors = returns_errors = return_error # cause I make too many typos
@return_error
def apply_with_return_error(args):
"""
args is a tuple where the first argument is a callable.
eg::
apply_with_return_error((func, 1, 2, 3)) - this will call func(1, 2, 3)
"""
return args[0](*args[1:])
