Python核心技术与实战——二十|assert的合理利用

我们平时在看代码的时候，或多或少会看到过assert的存在，并且在有些code review也可以通过增加assert来使代码更加健壮。但是即便如此，assert还是很容易被人忽略，可是这个很不起眼的用法，如果用的得当的话，会对我们的代码大有裨益。所以，我们今天就来看一看assert的用法。

什么是assert？

Python的assert可以被看做是一个debug的工具，主要测试一个条件是否满足，如果测试的条件满足，则什么也不执行，相当执行了pass语句；而如果条件不符合，则会抛出AssertionError，并返回具体的错误信息(optional)。他的具体语法是这样的

assert_stmt ::='assert' expression [',',Exception]

我们看看一个简单形式的assert expression的例子：

assert 1 == 2

就相当于下面的两行代码：

if __debug__:
    if not expression  : raise AssertionError

再开看看另外一种格式

assert 1 == 2,'assertion is wrong'

就相当于下面的两行代码格式

if __debut__:
    if not expression1:raise AssertionError(expression2)

这里的__debug__是一个常数，如果Python程序执行时候带了-o这个选项，比如

Python -O test.py 

那么，程序中所有assert语句都会失效，常数__debug__就为False，反之则为True。

不过要注意的是，直接对常数赋值是非法的，因为他的值在解释器运行的时候已经决定了，中途是无法改变的。

另外还要记住一点：在使用assert的时候一定不要加括号，比如下面的形式

assert (1 == 2,'this should fail')

因为这样写的话，无论表达式是否正确，assert检查都不会是fail，程序只会给出SyntaxWarning的消息，正确的写法是这样的

assert 1 ==2 ,'this should fail'
##########输出##########
AssertionError: this should fail

总的来说，asssert在程序中的作用，是对代码做一些internal的self-check。使用assert，就表示很确定这个条件一定会发生或者一定不会发生。

举个例子，比如这里有一个函数需要给定的参数是人的性别，而正常情况性别之分男和女，我们就可以使用assert来防止程序的非法输入。如果程序没有bug那么assert永远不会抛出异常，而一旦抛出异常就很容易定位程序是在哪里出了问题，便于定位。

assert的用法

讲完了assert的基本语法和概念，我们下面通过一些实际的场景来看看assert在Python中的用法，并弄清楚assert的使用场景。

第一个例子。我们假设超时在做促销活动，准备对一些商品进行打折，那么后台就需要做一个apply_discount()的函数，要求输入为原来的价格和折扣，输出是折扣以后的价格，那么，我们就可以大概写成下面的样子：

def apply_discount(price,discount):
    update_price = price*(1-discount)
    assert 0 < update_price'price should be greated or equal to 0'
    return update_price

然后我们可以通过给定几组数来验证一下他的功能

print(apply_discount(100,0.8))
##########输出##########
19.999999999999996


print(apply_discount(100,2))
##########输出##########
AssertionError: price should be greated or equal to 0

可以看出来，如果discount是0.2的时候，输出是正常的，但是discount如果成了2时，程序就会抛出异常了。这个时候，开发人员修改相关代码或增加进新功能的时候，在运行测试的时候非常容易看出问题。所以，assert的加入，可以有效的预防bug的发生，提高程序的健壮性。

第二个例子，最常见的除法操作是在哪个领域都会遇到的，比方我们想知道一批货物的销售的平均价格，就要给定销售总额还有销售数，这样平均售价就能算出来。

def calculate_average_price(totle_sales,num_sales):
    assert num_sales > 0,'number of sales should be greater than 0'
    return totle_sales/num_sales

同样我们在函数里增加了assert语句，规定了销售的总数量必须大于0，这样就可以防止后台计算错误。

除了上面两个例子，在实际工作中，assert还有一些很常见的用法，比方下面的场景：

def fun(input):
    assert isinstance(input,list),'input must be type of list'
    if len(input) == 1:
        pass
    elif len(input) == 2:
        pass
    else:
        pass

函数里的操作是基于input是个list这个前提，我们函数开始的地方加一个assert检查，防止程序出错。

但是要注意的是，在这个函数中加了assert的前提是我们十分确定程序的输入是一个list，而不能是其他的数据类型。如果我们的这个函数是个多态的，针对不同的数据类型有不同的操作，那就应该写成if...else...的条件语句了

def fun(input):
    if isinstance(input,int):
        pass
    elif isinstance(input,str):
        pass
    else:
        pass

assert的错误示例

通过前面讲的assert的使用场景，有可能会让我们比较迷茫——很多地方都可以使用assert，那么很多if...else的条件语句是不是也可以换成assert呢？这种想法可能就不准确了。接下来，我们看一看几个典型的错误用法：

比方我们要删除一些数据，但是删除操作必须是admin用户才可以，那么就有了下面的代码，

def delete_data(user,data_id):
    assert user_is_admin(user),'user must  admin'
    assert data_exist(data_id),'data must  exist'
    delete(data_id)

那么上面的代码有什么问题么？

assert的检查是可以被关闭的，在运行Python程序时候，加入一个-O选项就会使assert失效。因此，一旦assert的检查被关闭，user_is_admin()和course_exist()这两个函数就不会执行，就会导致下面的问题：

1.任何用户都可以删除数据；

2.不管数据是否存在，都可以墙纸执行删除操作

这就会给程序带来巨大的安全漏洞，正确的做法，是使用条件语句进行相应的检查，然后抛出相关的异常信息。

def delete_data(user,data_id):
    if not user_is_admin(user):
        raise Exception('user must be admin')
    if not data_exist(data_id):
        raise Exception('data must exist')
    delete(data_id)

再来看一个例子，我们想打开一个文件，进行数据的处理，读取等一系列操作，那么下面的写法也是有风险的

def read_and_process(path):
    assert file_exist(path),'file must exist'
    with opne(path) as f:
        pass

因为assert的使用，表明强行指定文件必须存在，但事实情况下，这个假设是不成立的，另外，打开文件操作也可能触发别的异常。所以正确的做法是用try...except来解决

def read_and_process(path):
    try:
        with open(path) as f:
            pass
    except Exception as e:
        pass

总得来说，assert是不实用于run-time error的检查，比方试图打开一个文件，但文件不存在；或者想要从网上down一个文件，但是中途断网了等。这些情况下我们一般使用错误和异常处理。

总结

我们今天学习了assert的用法——assert通常用来对代码进行必要的self-check，表明我们在写代码的时候很确定这种情况一定会发生，或者一定不会发生。需要注意的是，使用assert的时候，一定不能加括号，否则无论表达式对与错，assert的检查永远不会fail。另外，程序中的assert，可以通过-O等选项被全局disable。

通过几个场景的应用，我们可以发现assert的合理使用可以增加代码的健壮度，同时也方便了程序出错时开发人员的定位排查。

不过，我们也要注意assert的使用场合，大多数情况下程序中出现的不同情况都是意料之中的，西药我们用不同的方案来处理，这时候条件语句就更加合适，而程序中的一些run-time error，异常处理就更加合适