从堆栈跟踪中隐藏装饰器的 Python 2 和 3 兼容方式

Question

我写了一个不想出现在堆栈跟踪中的装饰器。所以在 Python2 中我会这样做：

class SneakyDecorator(object):
    def __init__(self, f):
        self.f = f

    def __call__(self, *args, **kwargs):
        # [...]
        try:
            self.f(*args, **kwargs)
        except:
            t, v, tb = sys.exc_info()
            raise v, None, tb.tb_next             # <=== Important line

在 Python3 中我会这样做：

class SneakyDecorator:
    def __init__(self, f):
        self.f = f

    def __call__(self, *args, **kwargs):
        # [...]
        try:
            self.f(*args, **kwargs)
        except:
            t, v, tb = sys.exc_info()
            raise v.with_traceback(tb.tb_next)   # <=== Important line

所以问题来了：有什么方法可以同时兼容 Python2 和 Python3？ 我更喜欢不涉及两个独立代码库的解决方案。

我尝试使用six 模块的reraise 函数，但问题是，该函数随后出现在堆栈跟踪中。

如果你在这个答案中这样做，同样的问题：Exception with original traceback - 2.6-3.X compatible version。

更新：Python3 代码不起作用！调用方法仍然显示在堆栈跟踪中。所以接下来的问题是：有没有办法在 Python3 中做到这一点？

Answer 1

实现这项工作的唯一方法是保持__call__ 方法的不同版本，因为exec 本身会创建一个堆栈帧！以下，在 Python 2 中再次揭示了框架：

class SneakyDecorator:
    def __init__(self, f):
        self.f = f

    def __call__(self, *args, **kwargs):
        # [...]
        try:
            self.f(*args, **kwargs)
        except:
            t, v, tb = sys.exc_info()
            exec("raise v, None, tb.tb_next")

结果：

>>> f()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<string>", line 12, in __call__
  File "<stdin>", line 3, in f
ValueError: Oi!

所以使用：

def __call__(self, *args, **kwargs):
    # [...]
    try:
        self.f(*args, **kwargs)
    except:
        t, v, tb = sys.exc_info()
        if sys.version_info[0] <= 2:
            exec("raise v, None, tb.tb_next")
        else:
            raise v.with_traceback(tb.tb_next)

实际上不会起作用。

对于 Python 2，以下内容与您的两个单独的类具有完全相同的功能：

if sys.version_info[0] <= 2:
    _call = '''\
def _call(self, *args, **kwargs):
    # [...]
    try:
        self.f(*args, **kwargs)
    except:
        t, v, tb = sys.exc_info()
        raise v, None, tb.tb_next
'''
    exec(_call)
else:
    def _call(self, *args, **kwargs):
        # [...]
        try:
            self.f(*args, **kwargs)
        except:
            t, v, tb = sys.exc_info()
            raise v.with_traceback(tb.tb_next)


class SneakyDecorator:
    def __init__(self, f):
        self.f = f

    __call__ = _call
    # niceties, patch up name and qualified name. Optional.
    __call__.__name__ = '__call__'
    __call__.__qualname__ = 'SneakyDecorator.__call__'

但是，请注意，对于 Python 3，Exception.with_traceback() 方法可能允许您附加一个新的回溯并删除当前帧，但 Python 会在您重新引发时将其重新添加例外！