如何使用: 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

Answer 2

import warnings
# import pdb

def format_Warning(message, category, filename, lineno, line=''):
    return str(filename) + ':' + str(lineno) + ': ' + category.__name__ + ': ' +str(message) + '\n'

warnings.formatwarning = format_Warning

def tt():

    warnings.warn("123")

    return 10

x = tt()

print(x)

希望它能帮上忙！