Python 捕获警告

来源 | 写代码的明哥

责编 | 寇雪芹

头图 | 下载于视觉中国

警告不是异常

你是不是经常在使用一些系统库或者第三方模块的时候，发现一些既不是异常也不是错误的警告信息？这些经常出现的警告信息，容易让新手造成一些误判，以为是程序出错了。

实则不然，异常和错误都是程序出现了一些问题，但是警告不同，他的紧急程度非常之低，以致于大多数的警告都是可以直接忽略的。如果不想显示这些告警信息，可以直接加上参数 -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)

运行后，效果如下

更多精彩推荐

☞用 Python 动态可视化，看看比特币这几年☞小小几张图，把深度学习讲透彻☞Python 搭建车道智能检测系统☞网易伏羲论文入选 CVPR：AI 感知表情能力或实现巨大突破！

点分享点收藏点点赞点在看