with 用法理解

with 用法理解

Overview

with 与with之后的object一起，起到了抛出异常和单独生成一个空间让代码在空间里运行的效果。

实验代码

class A:
    def __init__(self):
        self.a = 0

    def __enter__(self):
        print('enter')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('exit')


if __name__ == '__main__':

    a = A()
    with a:
        print('first step')
        print(1/0)
        print('last setp')
    print('continue running')
    print('continue running')
    print('continue running')
    print('continue running')

上述代码输出结果为

enter
first step
exit
ZeroDivisionError: division by zero

代码理解
根据上述代码的测试结果可以看出：

1. with语句先运行，with之后对象的__enter__()方法
2. 然后运行with空间的代码
   1.1. 当with空间代码出错后，会直接运行__exit__()方法，然后抛出异常
   1.2 当with空间代码没有错误时，程序按顺序__enter()__>> 逻辑语句>>exit()>>之后的代码继续运行