如何编写与 Python 2 和 Python 3 兼容的异常重新引发代码？

Question

我正在尝试使我的 WSGI 服务器实现与 Python 2 和 Python 3 兼容。我有以下代码：

def start_response(status, response_headers, exc_info = None):
    if exc_info:
        try:
            if headers_sent:
                # Re-raise original exception if headers sent.
                raise exc_info[0], exc_info[1], exc_info[2]
        finally:
            # Avoid dangling circular ref.
            exc_info = None
    elif headers_set:
        raise AssertionError("Headers already set!")

    headers_set[:] = [status, response_headers]
    return write

...相关部分是：

# Re-raise original exception if headers sent.
raise exc_info[0], exc_info[1], exc_info[2]

Python 3 不再支持该语法，因此必须将其翻译为：

raise exc_info[0].with_traceback(exc_info[1], exc_info[2])

问题：Python 2 语法在 Python 3 中生成解析错误。如何编写可以被 Python 2 和 Python 3 解析的代码？我已经尝试了以下方法，但这不起作用：

if sys.version_info[0] >= 3:
    raise exc_info[0].with_traceback(exc_info[1], exc_info[2])
else:
    eval("raise exc_info[0], exc_info[1], exc_info[2]; 1", None, { 'exc_info': exc_info })

Answer 1

你能用six吗？它的存在就是为了解决这个问题。

import six, sys
six.reraise(*sys.exc_info())

见：https://six.readthedocs.io/index.html#six.reraise

Answer 2

你可以做一些有创意的事情。

在代码的开头进行检查 - 你的构造函数或其他什么，检查你使用的 python 版本，因为你的普通版本检查器不起作用，试试这个：

try:
  eval('a python 3 expression') # something that only works in python3+
  python_version = 3
except:
  python_version = 2

然后你的代码的其余部分可以很容易地引用它来知道要使用什么。

至于解析错误，你可以在函数中使用 exec，如下所示：

def what_to_run():
    if python_version = 3:
        return 'raise exc_info[0].with_traceback(exc_info[1], exc_info[2])'
    else:
        return 'raise exc_info[0], exc_info[1], exc_info[2]'

在你的函数中你会这样写：

def start_response(status, response_headers, exc_info = None):
    if exc_info:
        try:
            if headers_sent:
                # Re-raise original exception if headers sent.
                exec(what_to_run())
        finally:
            # Avoid dangling circular ref.
            exc_info = None
    elif headers_set:
        raise AssertionError("Headers already set!")

    headers_set[:] = [status, response_headers]
    return write

有点乱，未经测试，但它应该工作，至少你理解这个想法。