python使用@contextmanager来定义上下文管理器

python 上下文管理器

上下文管理器是在Python2.5之后加入的功能，可以在方便的需要的时候比较精确地分配和释放资源, with便是上下文管理器的最广泛的应用; 比使用try:…finally:f.close方便的多.

with open("test/test.txt","w") as f:
	  f.write("hello")

上下文管理协议：实现了__enter__和__exit__方法。（一如迭代器协议是实现__iter__方法）
例：

class Contextor:
    def __enter__(self):
        print(1111)
        
    def __exit__(self, exc_type, exc_val, exc_tb):
        print(333)
        
	def test():
		print(222)

比如：

我们正常不使用用上下文的代码:

c = Contextor()
c.test()
# 结果如下：
>>> 222

当我们使用上下文的时候

with Contextor() as c:
	c.test()

# 结果如下
>>> 1111
>>> 222
>>> 333

类中有两个特殊的魔术方法:
enter: with语句中的代码块执行前, 会执行__enter__, 返回的值将赋值给with句中as后的变量.
exit: with语句中的代码块执行结束或出错, 会执行_exit__

如果我们要定义上下文管理器，就需要在这类里面定义__enter__和__exit__这两个方法。

有没有什么简化的方法不需要我们直接定义__enter__和__exit__呢？

python提供了一个装饰器叫@contextmanager，是位于contextlib模块下的。
借助contextmanager装饰器，我们可以简化MyResource的定义

现在我们不要__enter__和__exit__方法，只保留核心方法query()，注意我们只是不想在定义__enter__和__exit__这两个方法，但是他们里面所执行的语句我们还是需要实现的。在进入上下文管理器的时候打印__enter__里面的方法，在退出的时候打印__exit__里面的方法。

from contextlib import contextmanager


class Contextor:
    def test(self):
        print(222)

    @contextmanager
    def with_test(self):
        print(1111)
        yield self
        print(333)


with Contextor().with_test() as c:
    c.test()

# 结果如下
>>> 1111
>>> 222
>>> 333

被装饰器装饰的函数分为三部分:
with语句中的代码块执行前执行函数中yield之前代码
yield返回的内容复制给as之后的变量
with代码块执行完毕后执行函数中yield之后的代码

---

参考博文：
https://blog.csdn.net/qq_25473157/article/details/87103840
https://blog.csdn.net/JENREY/article/details/86690018