In Python, come si catturano gli avvisi come se fossero eccezioni?

Una libreria di terze parti (scritta in C) che utilizzo nel mio codice Python emette avvisi. Voglio essere in grado di utilizzare la try exceptsintassi per gestire correttamente questi avvisi. C'è un modo per fare questo?

Per citare dal manuale di python ( 27.6.4. Avvertenze sui test ):

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)

Per gestire gli avvisi come errori, usa semplicemente questo:

import warnings
warnings.filterwarnings("error")

Dopo questo sarai in grado di rilevare gli avvisi come gli errori, ad esempio questo funzionerà:

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

PS Aggiunta questa risposta perché la migliore risposta nei commenti contiene errori di ortografia: filterwarnignsinvece di filterwarnings.

Se vuoi solo che lo script fallisca sugli avvisi, puoi usare:

python -W error foobar.py

Ecco una variazione che rende più chiaro come lavorare solo con i tuoi avvisi personalizzati.

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)

In alcuni casi, è necessario utilizzare ctypes per trasformare gli avvisi in errori. Per esempio:

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