简介
python中存在这么一个非常好用的东西,能够帮助我们更好的管理上下文,这种东西就成为上下文管理器,例如我们最常见的with open,在读取文件时可以很方便的进行文件描述符和文件对象的打开和关闭,防止溢出,下面我们就介绍以下上下文管理器的实现的两个方式。
使用内置方法实现
实现__enter__
和__exit__
方法,注意的时__enter__
返回的是self,即返回的对象,__exit__
方法无论是否出现问题都会执行,类似与try-except中的finally。
class myContextManager:
def __enter__(self):
print("__enter__")
return self
# exec_val:异常种类, exec_val:异常值, exec_traceback:异常的堆栈
def __exit__(self, exec_type, exec_val, exec_traceback):
print(exec_val)
print("-------------")
print(exec_type)
print("-------------")
print(exec_traceback)
print("__exit__")
with myContextManager() as f:
raise IndexError
使用contextManager装饰器实现
使用contextManager
装饰器需要注意,必须要使用yield
,yield
前类似于__enter__
,yield返回的对象就是__enter__
返回的对象,yield后便是类似于__exit__
的方法
@contextmanager
def some_generator(<arguments>):
<setup>
try:
yield <value>
finally:
<cleanup>
对于有些上下文在出现异常后需要主动关闭时,需要主动关闭上下文对象,使用到closing
with closing(<module>.open(<arguments>)) as f:
<block>
总结
- 上下文管理器可以帮助我们做好管理,例如打开关闭数据库动作,