Trong Python, làm cách nào để bắt các cảnh báo như thể chúng là ngoại lệ?

Thư viện của bên thứ ba (được viết bằng C) mà tôi sử dụng trong mã python của mình đang đưa ra cảnh báo. Tôi muốn có thể sử dụng try exceptcú pháp để xử lý đúng các cảnh báo này. Có cách nào để làm việc này không?

Để trích dẫn từ cẩm nang python ( 27.6.4. Cảnh báo thử nghiệm ):

import warnings

def fxn():
    warnings.warn("deprecated", DeprecationWarning)

with warnings.catch_warnings(record=True) as w:
    # Cause all warnings to always be triggered.
    warnings.simplefilter("always")
    # Trigger a warning.
    fxn()
    # Verify some things
    assert len(w) == 1
    assert issubclass(w[-1].category, DeprecationWarning)
    assert "deprecated" in str(w[-1].message)

Để xử lý cảnh báo là lỗi, chỉ cần sử dụng điều này:

import warnings
warnings.filterwarnings("error")

Sau đó, bạn sẽ có thể nhận được các cảnh báo giống như lỗi, ví dụ như điều này sẽ hoạt động:

try:
    some_heavy_calculations()
except RuntimeWarning:
    import ipdb; ipdb.set_trace()

PS Đã thêm câu trả lời này vì câu trả lời hay nhất trong các nhận xét chứa lỗi chính tả: filterwarnignsthay vì filterwarnings.

Nếu bạn chỉ muốn tập lệnh không thành công khi có cảnh báo, bạn có thể sử dụng:

python -W error foobar.py

Đây là một biến thể giúp bạn hiểu rõ hơn về cách hoạt động chỉ với các cảnh báo tùy chỉnh.

import warnings
with warnings.catch_warnings(record=True) as w:
    # Cause all warnings to always be triggered.
    warnings.simplefilter("always")

    # Call some code that triggers a custom warning.
    functionThatRaisesWarning()

    # ignore any non-custom warnings that may be in the list
    w = filter(lambda i: issubclass(i.category, UserWarning), w)

    if len(w):
        # do something with the first warning
        email_admins(w[0].message)

Trong một số trường hợp, bạn cần sử dụng ctypes để biến cảnh báo thành lỗi. Ví dụ:

str(b'test')  # no error
import warnings
warnings.simplefilter('error', BytesWarning)
str(b'test')  # still no error
import ctypes
ctypes.c_int.in_dll(ctypes.pythonapi, 'Py_BytesWarningFlag').value = 2
str(b'test')  # this raises an error