目录
上下文管理对象
当一个对象同时实现了__enter__()和__exit__()方法,它就属于上下文管理的对象
1. __enter__
进入与此对象相关的上下文。如果存在该方法,with语法会把该方法的返回值作为绑定到as子句中指定的变量上
2.__exit__
退出与此对象相关的上下文
class Test:
def __init__(self):
print("init")
def __enter__(self):
print("enter")
def __exit__(self, exc_type, exc_val, exc_tb):
print("exit")
with Test() as f:
print("test")
说明:
- 实例化对象的时候,并不会调用__enter__,进入with语句块调用__enter__方法,然后执行语句体。最后离开with语句块的时候,调用 __exit__ 方法
- __enter__ 方法返回值就是上下文中使用的对象,with语法会把它的返回值赋给as子句的变量。
class Test:
def __init__(self):
print("init")
def __enter__(self):
print("enter")
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print("exit")
T = Test()
with T as f:
print(T == f)
print("test")
注:with可以开启一个上下文运行环境,在执行前做一些准备工作,执行后做一些收尾工作
3. __enter__方法和__exit__方法的参数
__enter__方法没有其他参数
__exit__方法有3个参数
__exit__ (self,exc_type, exc_value,traceback)这三个参数都与异常有关。
如果该上下文退出时没有异常,这3个参数都为None。如果有异常,参数意义如下:
exc_type,异常类型
exc_value,异常的值
traceback,异常的追踪信息
__exit__方法返回一个等效True的值,则压制异常;否则,继续抛出异常
class Test:
def __init__(self):
print("init")
def __enter__(self):
print("enter")
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print(exc_type)
print(exc_val)
print(exc_tb)
print("exit")
return "exit test"
T = Test()
with T as f:
raise Exception('test error')
print("test")
print("end")
上下文管理应用
- 增强功能:在代码执行的前后增加代码,以增强其功能。
- 资源管理:类似装饰器的功能。打开了资源需要关闭,例如文件对象、网络连接、数据库连接等
- 权限验证:在执行代码之前,做权限的验证,在__enter__中处理