python 上下文管理(with、contextmanager)

简介

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>

总结

• 上下文管理器可以帮助我们做好管理，例如打开关闭数据库动作，