作者: 一去、二三里
个人微信号: iwaleon
微信公众号: 高效程序员
Python 提供了丰富的标准异常,这些异常非常灵活,我们甚至可以根据需要进行修改以满足特定需求,所以应该尽可能地使用它们。
然而,在实际编程中,你会发现有时标准异常不够用,这就需要我们创建自定义异常来满足特殊的业务场景。
自定义异常
创建自定义异常,可以通过创建一个新类来实现,这个新类必须从 Exception
类直接或间接地派生出来。
注意: 大多数标准异常也是 Exception
类的派生形式,可参见:内置异常的层次结构。
例如,创建一个派生自 Exception
的自定义异常:
>>> class CustomException(Exception):
... pass
...
与其他异常一样,可以使用 raise
语句来引发此异常:
>>> raise CustomException
Traceback (most recent call last):
...
__main__.CustomException
>>>
>>> raise CustomException("A custom exception occurred")
Traceback (most recent call last):
...
__main__.CustomException: A custom exception occurred
实现异常类,比较好的做法是:将所有自定义异常放在一个单独的文件中(例如:exceptions.py
或 errors.py
),许多标准模块也都是这样做的。
既然自定义异常是类,那么它必然可以实现一个普通类能做的所有事情。但一般而言,应该尽量保持简单、简洁。大多数实现都是声明一个自定义基类,并从这个基类派生出其他的(由程序引发的)异常类。这是 Python 中实现自定义异常的标准方法,但并不仅限于这种方式。
实际案例
基于此,再完善下 Python 异常处理 中关于薪资的问题。输入一个数字,直到正确地猜测到薪资为止。为了解决这个问题,提供一些猜测大于/小于的提示。
定义了一个名为 CustomException
的基类,其他两个异常(ValueTooSmallException
和 ValueTooLargeException
)都派生自该类,将它们放入 exceptions.py
模块中:
class CustomException(Exception):
"""Base class for other exceptions"""
pass
class ValueTooSmallException(CustomException):
"""Raised when the input value is too small"""
pass
class ValueTooLargeException(CustomException):
"""Raised when the input value is too large"""
pass
然后,编写一个猜薪资的主程序 - guess_salary.py
:
from exceptions import *
# 需要猜测的薪资
salary = 10000
while True:
try:
i_salary = int(input("Please enter the salary: "))
if i_salary < salary:
raise ValueTooSmallException
elif i_salary > salary:
raise ValueTooLargeException
except ValueTooSmallException:
print("Too small, try again!\n")
except ValueTooLargeException:
print("Too large, try again!\n")
except:
print("It's not a number!\n")
else:
print("Congratulations!")
break
考虑到程序的健壮性,除了自定义异常之外,我们还捕获了其他异常。这样以来,即使输入的内容无法转换为 int,程序也不至于出错。
运行主程序,并输入相关值:
Please enter the salary: abc
It’s not a number!Please enter the salary: 5000
Too small, try again!Please enter the salary: 20000
Too large, try again!Please enter the salary: 10000
Congratulations!
完美,一切正如我们所愿!