1. 概述
这两个函数与with as语句语句紧密相连,即
class Test():def __enter__(self):print("Enter!")passdef __exit__(self, type, value, trace):print("Exit!")passwith Test() as f:
# 执行with as语句时,Test类中的__enter__函数会被调用,__enter__函数返回的对象成为fprint("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__函数返回的对象成为fprint("f:", f)a = 1 / 0print("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!