Effective Python - 函数

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