python - 如何使用 : python warnings. warn()？

Question

这是我写的代码:

#usr/bin/python3
import warnings

def tt():

    warnings.warn("123")

    return 10

x = tt()

print(x)

它打印:

test.py:5: UserWarning: 123

warnings.warn("123")

10

我希望它只打印:

test.py:5: UserWarning: 123
10

没有 warnings.warn("123")。

我该怎么办？

Answer 1

您可以替换用于格式化消息的函数；例如:

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

Python Module of the week article