python - 是否有可能捕获 python 中 try block 中引发的所有警告？

Question

我有一些代码大致如下:

from logging import warning

def complex_function():
    # Do some stuff
    raise Warning("blah") 
    # Do some more stuff
    raise Warning("Blah again") 

try: 
    complex_function()
except Warning as e: 
    warning(e) 

这会导致:

WARNING:root:blah

我想捕获所有发出的警告并记录它们。在我的代码中，此类警告有时来自第 3 方库，因此修改警告以使用 logging.warning 是不切实际的，而且我还想存储警告信息，以便我可以通过 API 返回该信息的某些版本。

有没有办法让我做这样的事情来捕获所有警告并循环它们？

编辑

太晚了，我意识到我在上面的示例中提出了错误的警告，并且 complex_function 应该是长行:

def complex_function():
    # Do some stuff
    warnings.warn("blah") 
    # Do some more stuff
    warnings.warn("Blah again", UnknownWarningType)

我想我可以用 warnings.catch_warnings 捕获这些内容

Answer 1

您是否期待类似以下内容:

import warnings
warnings.filterwarnings('error')


def warning_func():
    print('hello')
    warnings.warn(Warning('Warn1'))
    print('hi')
    warnings.warn(Warning('Warn2'))


with warnings.catch_warnings(record=True) as w:
    warnings.simplefilter("always")
    warning_func()
    print(w)