Python之__enter__和__exit__

1. 概述

这两个函数与with as语句语句紧密相连，即

class Test():

    def __enter__(self):
        print("Enter!")
        pass

    def __exit__(self, type, value, trace):
        print("Exit!")
        pass
    
with Test() as f:       
# 执行with as语句时，Test类中的__enter__函数会被调用，__enter__函数返回的对象成为f
    print("f:", f)
    print("Have a try!")

在以上例子中，执行顺序如下

• 执行with as语句时，Test类中的__enter__函数会被调用，因此首先会打印"Enter!"
• __enter__函数返回的对象成为f，这里__enter__函数没有return语句，默认返回None，因此f就是None
• 然后打印"Have a try!"
• 当with as语句执行完毕以后，__exit__函数会被调用，因此最后打印"Exit!"

所以最后的执行结果如下，

Enter!
f: None
Have a try!
Exit!

2. __exit__函数的三个参数

我们注意到__exit__函数中除了self参数外，还有三个参数type, value, trace，这三个参数有什么用呢？这三个参数是处理异常的，也就是在执行with as语句块中，如果遇到异常，会马上进入__exit__函数。例如，

class Test():

    def __init__(self):
        print("Init!")

    def __enter__(self):
        print("Enter!")
        return "F"

    def __exit__(self, type, value, trace):
        print("type:", type)
        print("value:", value)
        print("trace:", trace)
        print("Exit!")
        return True
        # 这里返回True意味着with as语句块后面的代码会继续执行
        # 但是with as语句块内部，产生异常之后的代码不会执行


with Test() as f:  # 执行with as语句时，Test类中的__enter__函数会被调用，__enter__函数返回的对象成为f
    print("f:", f)
    a = 1 / 0
    print("Have a try!")  # 猜猜这句话会不会执行呢？

print("Done!")  # 猜猜这句话会不会执行呢？

• 执行with as语句时，Test类中的__enter__函数会被调用，因此首先会打印"Enter!"
• __enter__函数返回的对象成为f，这里__enter__函数有return语句，返回F，因此f就是F
• 执行a = 1 / 0会出现异常，直接进入__exit__函数，打印出相应的type，value，traceback和"Exit!"
• 由于已经进入__exit__函数，with as语句块中后面的内容将不再执行，因此不会打印"Have a try!"

注意，在__exit__函数返回了True，这意味着with as语句块后面的内容将继续执行，因此会打印"Done!"
最后的执行结果如下

Init!
Enter!
f: F
type: <class 'ZeroDivisionError'>
value: division by zero
trace: <traceback object at 0x0000017F773D4288>
Exit!
Done!