能否从Python异常追踪中移除装饰器？

Question

我有几个装饰器，它们包裹了很多函数。当其中一个函数出错时，我的错误追踪信息里会出现很多装饰器的相关信息，这让我觉得很烦。有没有办法修改一个Python装饰器，让它在出现异常时不在错误追踪信息中显示自己呢？

这里有一些示例代码：

import functools


def dec(func):
    @functools.wraps(func)
    def wrap(*args, **kwargs):
        return func(*args, **kwargs)
    return wrap


@dec
def spam():
    print('I sure hope nobody throws an Exception')
    eggs()


@dec
def eggs():
    raise Exception('Spanish Inquisition')


if __name__ == '__main__':
    spam()

当我运行这个代码时，我得到的结果是：

Traceback (most recent call last):
  File "tmp.py", line 23, in <module>
    spam()
  File "tmp.py", line 7, in wrap
    return func(*args, **kwargs)
  File "tmp.py", line 14, in spam
    eggs()
  File "tmp.py", line 7, in wrap
    return func(*args, **kwargs)
  File "tmp.py", line 19, in eggs
    raise Exception('Spanish Inquisition')
Exception: Spanish Inquisition

但我其实想要的是这样的结果：

Traceback (most recent call last):
  File "tmp.py", line 23, in <module>
    spam()
  File "tmp.py", line 14, in spam
    eggs()
  File "tmp.py", line 19, in eggs
    raise Exception('Spanish Inquisition')
Exception: Spanish Inquisition

我该怎么做呢？

Answer 1

我写了一个叫做 ignores_exc_tb 的装饰器，似乎能满足我的需求。下面我会把代码贴出来，还有我实际使用的代码：

import sys
from functools import wraps
from .util_iter import isiterable
from .util_print import Indenter


IGNORE_EXC_TB = not '--noignore-exctb' in sys.argv


def ignores_exc_tb(func):
    """ decorator that removes other decorators from traceback """
    if IGNORE_EXC_TB:
        @wraps(func)
        def wrapper_ignore_exctb(*args, **kwargs):
            try:
                return func(*args, **kwargs)
            except Exception:
                # Code to remove this decorator from traceback
                exc_type, exc_value, exc_traceback = sys.exc_info()
                # Remove two levels to remove this one as well
                raise exc_type, exc_value, exc_traceback.tb_next.tb_next
        return wrapper_ignore_exctb
    else:
        return func


def indent_decor(lbl):
    def indent_decor_outer_wrapper(func):
        @ignores_exc_tb
        @wraps(func)
        def indent_decor_inner_wrapper(*args, **kwargs):
            with Indenter(lbl):
                return func(*args, **kwargs)
        return indent_decor_inner_wrapper
    return indent_decor_outer_wrapper


def indent_func(func):
    @wraps(func)
    @indent_decor('[' + func.func_name + ']')
    @ignores_exc_tb
    def wrapper_indent_func(*args, **kwargs):
        return func(*args, **kwargs)
    return wrapper_indent_func


def accepts_scalar_input(func):
    '''
    accepts_scalar_input is a decorator which expects to be used on class methods.
    It lets the user pass either a vector or a scalar to a function, as long as
    the function treats everything like a vector. Input and output is sanatized
    to the user expected format on return.
    '''
    @ignores_exc_tb
    @wraps(func)
    def wrapper_scalar_input(self, input_, *args, **kwargs):
        is_scalar = not isiterable(input_)
        if is_scalar:
            iter_input = (input_,)
        else:
            iter_input = input_
        result = func(self, iter_input, *args, **kwargs)
        if is_scalar:
            result = result[0]
        return result
    return wrapper_scalar_input


def accepts_scalar_input_vector_output(func):
    '''
    accepts_scalar_input is a decorator which expects to be used on class
    methods.  It lets the user pass either a vector or a scalar to a function,
    as long as the function treats everything like a vector. Input and output is
    sanatized to the user expected format on return.
    '''
    @ignores_exc_tb
    @wraps(func)
    def wrapper_vec_output(self, input_, *args, **kwargs):
        is_scalar = not isiterable(input_)
        if is_scalar:
            iter_input = (input_,)
        else:
            iter_input = input_
        result = func(self, iter_input, *args, **kwargs)
        if is_scalar:
            if len(result) != 0:
                result = result[0]
        return result
    return wrapper_vec_output

编辑：
这是新的 Python 3 语法，用于重新抛出异常：

exc_type, exc_value, exc_traceback = sys.exc_info() try: exc_traceback = exc_traceback.tb_next exc_traceback = exc_traceback.tb_next except Exception: pass ex = exc_type(exc_value) ex.__traceback__ = exc_traceback raise ex