编写工具函数(utility function)时,python程序员喜欢给None这个返回值赋予特殊意义。这么做有时是合理的。例如,要编写辅助函数,计算两数相除的商。在除数为0的情况下,计算结果是没有明确含义的(undefinde,未定义的),所以似乎应该返回None.
def divide(a,b):
try:
return a/b
except ZeroDivisionError:
return None
函数的调用者,可以对这种特殊的返回值做出相应的解读。
result = divide(x,y)
if result is None:
print('Invalid inputs')
这样看,貌似没问题,但是如果调用的人想判断结果是不是零时,可能不会专门的判断返回值是否为None,可能会这样写代码:
x,y = 5,0
result = divide(x,y)
if not result:
print("Invalid inputs")
但这样有个问题,不管result返回的是0还是None,对于这段代码都是一样的。如果None对函数有特殊的意义,那么使用它的人就很容易写出错误的代码。有两种办法可以减少这种错误。
第一种办法,把返回值拆成两部分,并放到二元组(two-tuple)里面。二元组的首个元素,表示操作是否成功,接下来的那个元素,才是真正的运算结果。
def divide(a,b):
try:
return True, a/b
except ZeroDivisionError:
return False, None
调用该函数的人需要解析这个元组。这就迫使他们必须根据元组中表示运算状态的那个元素来做判断,而不能像从前那样,直接根据相除结果判断。
success,result = divide(x,y)
if not success:
print('Invalid inputs')
问题在于,调用者可以通过以下划线为名称的变量,轻易跳过元组的第一部分。这样写出 的代码,看上去似乎没错,但实际上,却和直接返回None的那种情况有着相同的错误。
_, result = divide(x,y)
if not result:
print('Invalid inputs')
第二种办法更好一些,那就是根本不返回None,而是把异常抛给上一级,使得调用者必须应对它。本例中,把ZeroDivisionError转化成ValueError, 用以表示调用者所给的输入值是无效的:
def divide(a,b):
try:
return a/b
except ZeroDivisionError as e:
raise ValueError('Invalid inputs') from e
现在,调用者就需要处理因输入值无效而引发的异常了( 这种抛出异常的行为,应该写入开发文档,参见该书第49条)。 调用者无需用条件语句来判断函数的返回值,因为如果函数没有抛出异常,返回值自然就是正确的。这样写出来的异常处理代码,它比较清晰。
要点:
- 用None 这个返回值来表示特殊意义的函数,很容易使调用都犯错误,因为None和0 及空字符串之类的值,在条件表达式里都 会评估为False.
- 函数在遇到特殊情况时,应该抛出异常,而不要返回None。 调用者看到该函数的文档中所描述的异常之后,应该就会编写相应的代码来处理它们了。