是否可以全局覆盖print()？

Question

我正在调试一个巨大的遗留代码库，实际上是将它从python2移植到python3。不幸的是，有一些异常似乎是可以处理和打印的，但是以前的开发人员没有足够的智慧来打印堆栈跟踪，它可以告诉我异常到底是在哪里触发的。

具体地说，我还将代码从GTK+2移植到GTK+3，我得到了许多这样的输出行：

TypeError: expose() missing 1 required positional argument: 'event'

在这一点上，似乎剩下的唯一选择是全局劫持print()函数，以至少显示它是在哪个模块中以及在哪一行被调用的。

你知道这在python3中是否可行吗？(我是python2的老手，但我是python3的新手)

关于如何解决这个问题，有什么其他建议吗？

Answer 1

您可以将sys.stdout和sys.stderr重新映射到一个被覆盖的类文件对象，每当有东西写入它时，检查您当前是否在异常框架内，如果是，则重新引发异常。

import io
import sys


class Stream(io.StringIO):

    def write(*args, **kwargs):
        if sys.exc_info()[0] is not None:
            raise
        super().write(*args, **kwargs)

stream = Stream()
sys.stdout = stream
sys.stderr = stream