python标准库--contextlib模块

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()