断言assert在异常中应用

发布时间 2023-07-22 12:17:46作者: Allen_Hao

语法&示例

 1 '''
 2 断言(assert)是一种用于检查程序中某个条件是否为真的语句。它用于在代码中插入自我验证的机制,以确保代码的正确性。
 3 1. 语法: assert condition, message
 4     condition是一个表达式,用于判断某个条件是否为真。如果条件为假,则断言失败,触发AssertionError异常。记住条件为False时,才触发异常哦
 5     message是可选的错误消息,用于描述断言失败的原因。
 6 
 7 2. 最佳实践
 8     1. 断言适用于开发和调试阶段:断言通常用于编写测试用例和进行调试,用于确保程序的正确性。
 9        在生产环境中,应该避免使用过多的断言,因为它们会影响程序的性能。
10     2. 使用断言来验证程序的前提条件和不变量:断言可以用来验证函数的输入参数、数据结构的状态或任何其他需要满足的前提条件。
11        它还可以用于验证程序执行过程中的不变量,即始终应该为真的条件
12 3. 坑
13     1. 不要滥用断言:断言应该用于检查预期情况下的条件,并且不应该用于处理用户输入或其他不可控的情况。如果条件失败,断言会触发异常,导致程序终止。
14     2. 断言不是异常处理机制:断言主要用于检查程序的正确性,而不是捕获和处理异常。因此,不应该将断言用于替代常规的错误处理机制。
15 '''
16 
17 
18 class Bank():
19     title = "zgyh Bank"
20 
21     def __init__(self, uname, money):
22         self.name = uname
23         self.balance = money
24 
25     def save_money(self, money):
26         assert money > 0, '存款money必须大于0'  # 真实开发,使用if语句去判断,然后raise抛异常
27         self.balance += money
28         print(f"{self.name}存款{money}完成")
29 
30     def withdrw_money(self, money):
31         assert money > 0, '取款money必须大于0'
32         assert money <= self.balance, '存款金额不足'
33         self.balance -= money
34         print(f"{self.name}取款{money}完成")
35 
36     def get_balance(self):
37         print(f"{self.name}目前余额:{self.balance}")
38 
39 
40 allen = Bank('Allen', 100)
41 allen.get_balance()
42 allen.save_money(300)
43 allen.get_balance()
44 allen.save_money(-300)  # 抛异常
45 allen.get_balance()  # 因异常未执行
46 '''
47 输出:
48     Allen目前余额:100
49     Allen存款300完成
50     Allen目前余额:400
51     Traceback (most recent call last):
52       File "F:\allen_class\python\base\base\038断言\038断言.py", line 44, in <module>
53         allen.save_money(-300)
54       File "F:\allen_class\python\base\base\038断言\038断言.py", line 26, in save_money
55         assert money > 0, '存款money必须大于0'  # 真实开发,使用if语句去判断,然后raise抛异常
56                ^^^^^^^^^
57     AssertionError: 存款money必须大于0
58 '''

停用断言

'''
在生产环境中,我们应该关闭断言(通过设置-O命令行选项或PYTHONOPTIMIZE环境变量),以避免不必要的性能开销。

关闭断言可以通过以下两种方式实现:
    1. 使用-O命令行选项:
        在运行Python脚本时,可以通过添加-O选项来关闭断言。这将导致解释器在执行期间忽略所有的assert语句。
        示例命令:python -O script.py

    2. 设置PYTHONOPTIMIZE环境变量:
        可以设置PYTHONOPTIMIZE环境变量为1来关闭断言。这样做会告诉解释器在执行期间忽略所有的assert语句。
        示例命令(Windows):set PYTHONOPTIMIZE=1
        示例命令(Linux/Mac):export PYTHONOPTIMIZE=1

请注意,关闭断言可能会提高代码的性能,但也会导致在程序出现问题时无法及时发现和调试。因此,应该谨慎使用,并确保在生产环境中启用了适当的错误处理机制。
'''