Python：异常装饰器。如何保存堆栈跟踪

Question

我正在编写一个装饰器来应用于函数。它应该捕获任何异常，然后根据原始异常消息引发自定义异常。 （这是因为 suds 抛出一个通用的 WebFault 异常，根据该异常的消息，我解析 Web 服务抛出的异常，并引发一个 Python 异常来镜像它。）

但是，当我在包装器中引发自定义异常时，我希望堆栈跟踪指向引发原始 WebFault 异常的函数。到目前为止，我所拥有的引发了正确的异常（它动态解析消息并实例化异常类）。 我的问题：如何保留堆栈跟踪以指向引发 WebFault 异常的原始函数？

from functools import wraps

def try_except(fn):
        def wrapped(*args, **kwargs):
            try:
                fn(*args, **kwargs)
            except Exception, e:
                parser = exceptions.ExceptionParser()
                raised_exception = parser.get_raised_exception_class_name(e)
                exception = getattr(exceptions, raised_exception)
                raise exception(parser.get_message(e))
        return wraps(fn)(wrapped)

Answer 1

在 Python 2.x 中，raise 的一个鲜为人知的特性是它可以与多个参数一起使用：raise 的三参数形式接受异常类型、异常实例和回溯。您可以使用 sys.exc_info() 获取回溯，它返回（并非巧合）异常类型、异常实例和回溯。

（将异常类型和异常实例视为两个单独的参数的原因是异常类出现之前的产物。）

因此：

import sys

class MyError(Exception):
    pass

def try_except(fn):
    def wrapped(*args, **kwargs):
        try:
            return fn(*args, **kwargs)
        except Exception, e:
            et, ei, tb = sys.exc_info()
            raise MyError, MyError(e), tb
    return wrapped

def bottom():
   1 / 0

@try_except
def middle():
   bottom()

def top():
   middle()

>>> top()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "tmp.py", line 24, in top
    middle()
  File "tmp.py", line 10, in wrapped
    return fn(*args, **kwargs)
  File "tmp.py", line 21, in middle
    bottom()
  File "tmp.py", line 17, in bottom
    1 / 0
__main__.MyError: integer division or modulo by zero

在 Python 3 中，这发生了一些变化。在那里，回溯被附加到异常实例，并且它们有一个 with_traceback 方法：

raise MyError(e).with_traceback(tb)

另一方面，Python 3 也有异常链接，这在许多方面更有意义案例；要使用它，您只需使用：

raise MyError(e) from e

Answer 2

我在用我的自定义装饰器装饰的测试中遇到了这个问题。

我在装饰器主体中使用了以下构造来保留在单元测试输出中打印的原始跟踪：

try:
    result = func(self, *args, **kwargs)
except Exception:
    exc_type, exc_instance, exc_traceback = sys.exc_info()
    formatted_traceback = ''.join(traceback.format_tb(
        exc_traceback))
    message = '\n{0}\n{1}:\n{2}'.format(
        formatted_traceback,
        exc_type.__name__,
        exc_instance.message
    )
    raise exc_type(message)