让Python的`warnings.warn()`不提及自己
问题内容:
我最小的例子是
#!/usr/bin/python3
import warnings
warnings.warn('Run Forest run!', stacklevel=2)
warnings.warn('Run Forest run!')
它会输出
sys:1: UserWarning: Run Forest run!
./file.py:6: UserWarning: Run Forest run!
warnings.warn('Run Forest run!')
第一行给我一点信息。第二行是完美的,为我提供了源文件和行号…但是我想摆脱多余的第三行。那可能吗?
问题答案:
事实证明,有可能让我们warnings.warn()
收集所有信息,而只是汇总信息的打印方式:
#!/usr/bin/python3
import warnings
def warning_on_one_line(message, category, filename, lineno, file=None, line=None):
return '%s:%s: %s: %s\n' % (filename, lineno, category.__name__, message)
warnings.formatwarning = warning_on_one_line
warnings.warn('Run Forest run!', stacklevel=2)
warnings.warn('Run Forest run!')
输出:
sys:1: UserWarning: Run Forest run!
./file.py:15: UserWarning: Run Forest run!
资料来源: 本周的Python模块