Effective Python - 函数
14.尽量用异常来表示特殊情况,而不要返回None
(1)如果令函数返回None来表示一些特殊意义,调用者很容易写出错误的代码,因为None,0,空字符串,空列表一类值,在条件表达式中会被判定为False。
def divide(x, y): try: return x / y except ZeroDivisionError: return None res1 = divide(0, 3) res2 = divide(3, 0) if not res1: print('invalid input') #invalid input if not res2: print('invalid input') #invalid input
说明:我们一般不会去判定函数的返回值是否为None,而是去假定:只要函数返回了与False等效的运算结果,就说明函数出错。
(2)函数在遇到特殊情况下应抛出异常,并在函数文档中添加相应的描述,不应返回None;
调用者在看到函数文档中关于异常的描述后,会编写相应的代码来处理它们。
def divide(x, y): try: return x / y except ZeroDivisionError as e: raise ValueError('invalid inputs') from e
说明:不返回None,而是把异常抛给上一级,使调用者必须应对它;
将ZeroDivisionError转化成ValueError用以表示调用者输入的值是无效的。
还有一种方法:把返回值拆成两部分放到元组里
def divide(x, y): try: return True, x / y except ZeroDivisionError: return False, None success, res = divide(3, 0) if not success: print('invalid input') #invalid input
但是这种方法存在一个问题,调用者可以通过下划线为名称的变量跳过元组的第一部分。
(python习惯用下划线来表示用不到的变量)
这和直接返回None有着相同的错误,所以并不推荐这种写法。
_, res = divide(0, 3) if not res: print('invalid input') #invalid input