python之上下文管理器详解

上下文管理器

定义：指实现上下文管理协议方法的对象
作用：主要用于释放资源（如打开的文件，网络连接，数据库连接，对对象的锁定等）

上下文管理协议的两个方法：
**1__enter__(self)**创建并返回一个可以引用的资源对象，供with语句块中的程序使用
**2__exit__(self，type，value，tb)**退出上下文时调用的，主要用来安全的释放资源

使用上下文管理器的with语句的形式为：

with context as val:
      pass

val将取得上下文管理器中__enter__()方法返回的资源引用供with后面的代码块使用

open函数的返回值

with open("filename", encoding="utf-8") as fp:
      fp.read()

此段代码中函数open()返回的是一个实现了上下文管理协议方法的对象，而并非是我们所认为的资源（文件句柄），这需要上下文管理器的__enter__（self）方法来返回文件句柄，供as后的变量fp引用。
我们可以利用python内置函数来检测一下open（）返回得对象得属性：
即：


open（）–>上下文管理器 --> enter（self）–> 资源

我们也可以实现一个自己的上下文管理器
代码如下：

class MyFileMgr:
   def __init__(self, fp=None, filename):
      self.fp = fp
      self.filename = filename
      
   def __enter__(self):  # 主要用于返回一个可以引用的资源
      self.fp = open(self.filename, encoding="utf-8")
      return self.fp
   def __exit__(self, t, v, tb):  # 用于释放资源
      if self.fp:
         fp.close()

if __name__ == '__main__':
    with MyFileMgr as fp:
         fp.read()

Py标准库中与上下文管理器相关的模块：Contextlib:
其提供了把生成器转换成上下文管理器的装饰器：contextmanager

import contextlib
@contextlib.contextmanager
def my_mgr(s, e):
    print(s)
    yield s + ' ' + e
    print(e)


if __name__ == "__main__":
    with my_mgr('start', 'end') as val:
        print(val)

运行效果：start
start end
end

代码调试得：

在此例中：
因为生成器转换成上下文管理器，所以yield下面的语句也会因为上下文管理器中的__exit__()释放资源，执行next(),从而执行输出end