python - 如何将 python 警告重定向到自定义流？

Question

假设我有一个类似 StreamIO 的文件对象，并希望 python 的警告模块将所有警告消息写入它。我怎么做？

Answer 1

尝试重新分配warnings.showwarning，即

#!/sw/bin/python2.5

import warnings, sys

def customwarn(message, category, filename, lineno, file=None, line=None):
    sys.stdout.write(warnings.formatwarning(message, category, filename, lineno))

warnings.showwarning = customwarn
warnings.warn("test warning")

将所有警告重定向到标准输出。

Answer 2

我认为这样的事情会起作用，尽管它是未经测试的代码，并且界面看起来有一种更清洁的方式目前让我无法理解：

import warnings

# defaults to the 'myStringIO' file
def my_warning_wrapper(message, category, filename, lineno, file=myStringIO, line=None):
    warnings.show_warning(message, category, filename, lineno, file, line)    

warnings._show_warning = my_warning_wrapper

如果这还不够的话，看看 Lib\warnings.py 应该可以帮助你走上正轨。

Answer 3

import sys
import StringIO

sys.stdout = StringIO.StringIO()