Python 使用过滤回溯引发异常

Question

我正在尝试使用不包含最后一帧的堆栈跟踪引发异常。

在 python 2 中，可以像这样从 strack 中删除最后一个条目：

def exceptionCatcher(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except Exception:
            excInfo = sys.exc_info()
            # some custom logging stuff
            raise excInfo[0], excInfo[1], excInfo[2].tb_next
    return wrapper

def a():
    raise Exception("My Exception")

@exceptionCatcher
def test():
    a()

这在 python 中有效，并从回溯中排除了最后一个 exceptionCatcher 调用本身，例如

Traceback (most recent call last):
  File "test.py", line 15, in <module>
    test()
  File "test.py", line 12, in a
    raise Exception("a")
Exception: a 

代替

Traceback (most recent call last):
  File "test.py", line 12, in <module>
    test()
  File "test.py", line 9, in wrapper
    raise e
  File "test.py", line 5, in wrapper
    return func(*args, **kwargs)
  File "test.py", line 10, in test
    a()
  File "test.py", line 5, in a
    raise Exception("a")
Exception: a

切换到 python3 不可能引发异常，例如： raise excInfo[0], excInfo[1], excInfo[2].tb_next

我不得不使用：

def exceptionCatcher(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except Exception as e:
            excInfo = sys.exc_info()
            # some custom logging stuff
            raise e.with_traceback(excInfo[2].tb_next)
...  

在我的回溯中添加了以下行：

Traceback (most recent call last):
  File "test.py", line 12, in <module>
    test()
  File "test.py", line 9, in wrapper
    raise e.with_traceback(excInfo[2].tb_next)
  File "test.py", line 10, in test
    a()
  File "test.py", line 5, in a
    raise Exception("a")
Exception: a

有什么办法可以摆脱这部分回溯？

我知道 exceptionCatching 可以用 exceptionhook 完成，但在我使用这个的场景中，不可能使用 exceptionhook，因为它是另一个第三方应用程序（在 cpp 中），它似乎捕获异常，所以 exceptionhook 不会捕获任何异常。 所以我想出了使用包装装饰器的想法，如上所示，尽管这是一种 hacky 方式。

Answer 1

使用上下文管理器代替装饰器，它首先不添加堆栈框架。

import contextlib


@contextlib.contextmanager
def exceptionCatcher():
    try:
        yield
    except Exception as e:
        print("custom logging")
        raise


def a():
    raise Exception("My Exception")


def test():
    with exceptionCatcher():
        a()


test()