第14条:尽量用异常来表示特殊情况,而不要返回Nono
核心知识点:
1.用None这个返回值来表示特殊意义的函数,很容易使调用者犯错,因为None和0以及空字符串之类的值,在条件表达式里都会评估为False。
2.两种方法:二元法;将异常抛给上一级直接报错。
编写工具函数(utility function)时,我们有时候喜欢给None这个返回值赋予特殊意义,有时候这么做是合理地,有时候就会出错。
例如要编写辅助函数,计算两数相除的商,在除数为0的情况下,计算结果是没有明确含义的,所以似乎应该返回None,就像下面这样:
In [1]: def divide(a,b): ...: try: ...: return a / b ...: except ZeroDivisionError: ...: return None ...: #貌似没有什么问题 #执行几次试试 In [2]: divide(1,2) Out[2]: 0.5 In [3]: divide(1,0) In [4]: divide(0,1) Out[4]: 0.0
貌似是没有什么问题,但是当在if语句中拿这个计算结果做判断时,会出现问题。
我们可能不会专门去判断函数的返回值是否为None,而是:只要返回与False等效的运算结果,就说明函数出错了。
这样明显有漏洞,当a=0,函数没有报错,但是按照上面的逻辑却是会报错。
如果None这个返回值,对函数有特殊意义(在上面的例子中,None的特殊意义就是异常),那么在编写Python代码时,就很容易犯上面这种错误。
有两种方法可以减少这种错误:
第一种方法,是把返回值拆成两部分,并放到二元组(two-tuple)里面。二元组的首个元素表示操作是否成功,接下来的那个元素,才是真正的运算结果。
In [9]: def divide(a,b): ...: try: ...: return True,a / b ...: except ZeroDivisionError: ...: return False, None ...: In [10]: divide(1,0) Out[10]: (False, None) In [11]: divide(0,1) Out[11]: (True, 0.0)
调用该函数的人需要解析这个元祖。这就使得不能只根据结果来遐想了,而必须根据运算状态的那个元素来做判断。
第二种办法更好一些,那就是根本不返回None,而是把异常把异常抛给上一级,使得调用者必须应对它。
In [13]: def divide(a,b): ...: try: ...: return a / b ...: except ZeroDivisionError as e: ...: raise ValueError('Invalid inputs') from e ...:
调用者需要处理因输入值无效而引发的异常。
调用者无需用条件语句来判断函数的返回值,因为如果函数没有抛出异常,返回值自然就是正确的,这样写出来的异常处理代码,也比较清晰。