python - 原始回溯的异常 - 2.6-3.X 兼容版本

Question

假设我有一个这样的上下文管理器——它在 Python 2.X 中工作并在退出时保留回溯。

class MyContextManager(object):
    def __enter__(self):
        return self
    def __exit__(self, exc_type, exc_value, traceback):
        raise AssertionError("Failed :-/"), None, traceback

在 Python 3 中， raise 是一个语法错误，但我认为你可以设置__traceback__参数。

def __exit__(self, exc_type, exc_value, traceback):
    e = AssertionError("Failed :-/")
    e.__traceback__ = traceback
    raise e

有没有办法保留与 Python 2 和 Python 3 兼容的回溯（即，两者都不会产生语法错误）？我在这一点上有点卡住了。它需要在 2.6、2.7、3.2 和 3.3 中工作。目标是确保用户仍能看到较早的回溯。

Answer 1

我想到的一个丑陋但可行的答案（受Ned Batchelder 的 Python 3 兼容性指南的启发）是编写一个函数来仅在 Python 2 时评估语法破坏代码。例如：

if sys.version_info[0] == 2:
   s = """
def raise_with_traceback(exc, traceback):
    raise exc, None, traceback
"""
   exec (s)
else:
   def raise_with_traceback(exc, traceback):
       raise exc.with_traceback(traceback)

[@user2357112 关于with_tracebackPython 3 首选的提示]。