Как перенаправить предупреждения 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 sys
import StringIO

sys.stdout = StringIO.StringIO()

Answer 3

Я думаю, что-то вроде этого будет работать, хотя это непроверенный код и интерфейс выглядит так, как будто есть более чистый способ, который ускользает от меня в настоящее время:

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, и вы окажетесь на правильном пути, если этого недостаточно.