以下文章来源于Python编程时光,作者写代码的明哥。
在前面我们介绍了如何在python中使用try except语句来测试代码。try except语句其实只能捕获异常或者错误,但无法捕获警告。在本文中,我向大家介绍几种捕获警告的方法。
当我们在使用第三方库的时候,经常会出现一些警告,数量往往还会很多。一些新手就会以为是不是我们的code写错了,code断了之类的。
但其实并不是这样的。异常和错误表明我们写的程序出现一些问题,但是对于警告而言,大多数的警告都是可以直接忽略的,它们的紧急程度非常之低。
如果不想显示这些告警信息,可以直接加上参数 -W ignore
参数,就不会再显示了。
能捕获的只有错误异常,但是我们可以通过一系列的操作将这些警告转化为异常。
这样的话,我们就可以像异常一样去捕获他们了。
如果我们不进行任何设置,警告就会直接打印在终端上。例子如下:
在 warnings 中有一系列的过滤器。
当你指定为 error 的时候,就会将匹配警告转换为异常。这样一来我们就可以通过异常的方式去捕获警告了。
import warnings
warnings.filterwarnings('error')
try:
warnings.warn("deprecated", DeprecationWarning)
except Warning as e:
print(e)
运行后,效果如下
如果你不想对在代码中去配置将警告转成异常。
import warnings
try:
warnings.warn("deprecated", DeprecationWarning)
except Warning as e:
print(e)
可以在执行的时候,只要加上一个参数 -W error
,就可以实现一样的效果
$ python3 -W error demo.py
deprecated
除了上面的方法之外 ,warnings 还自带了个捕获警告的上下文管理器。
当你加上 record=True
它会返回一个列表,列表里存放的是所有捕获到的警告,我将它赋值为 w
,然后就可以将它打印出来了。
import warnings
def do_warning():
warnings.warn("deprecated", DeprecationWarning)
with warnings.catch_warnings(record=True) as w:
do_warning()
if len(w) >0:
print(w[0].message)
运行后,效果如下