contextlib模块简介
- contextlib模块:实现上下文管理。
- 在Python中实现上下文管理是使用with关键字,但是默认只能实现特定的上下文管理。
- 其时,只要正确的实现
__enter__()
和__exit__()
魔术方法,就可以实现上下文管理。
class Context:
def __init__(self,bookname):
self.bookname = bookname
def __enter__(self):
print("开始上文管理")
return self
def __exit__(self, exc_type, exc_val, exc_tb):
if exc_type:
print("Error")
else:
print("实现下文管理")
def query(self):
return print(self.bookname)
with Context('wzry') as c:
c.query()
contextlib模块的装饰器
- @contextmanager:简写上下文管理方式
import contextlib
class Context:
def __init__(self,bookname):
self.bookname = bookname
def query(self):
print(self.bookname)
@contextlib.contextmanager
def d_query(name):
print("实现上文管理")
c = Context(name)
yield c
print("实现下文管理")
with d_query('wzry') as c:
c.query()