在Python中，如何捕獲警告，就好像它們是異常一樣？

Question

我在我的python代碼中使用的第三方庫（用C編寫）發出警告。我希望能夠使用tryexcept語法正確處理這些警告。有沒有辦法做到這一點？

Answer 1

要從蟒蛇手冊（27.6.4. Testing Warnings）引述：

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) 

（編輯：固定的例子，被關段）

Answer 2

這裏有一個變化，使得它更清楚如何在只自定義工作警告。

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) 

Answer 3

要處理warnnings爲錯誤簡單地使用這樣的：

import warnings 
warnings.filterwarnings("error") 

此之後，你將能夠趕上相同的錯誤，例如警告這將工作：

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

P.S.添加了此答案，因爲評論中的最佳答案包含拼寫錯誤：filterwarnigns而不是filterwarnings。

Answer 4

如果你只是想讓你的腳本失敗的警告，你可以使用：

python -W error foobar.py 

Answer 5

在某些情況下，你需要使用ctypes的把警告變爲錯誤。例如：

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