基于生成器实现上下文管理器(contextmanager装饰器)

　　上下文管理的实现除了靠定义类实现__enter__和__exit__ 方法之外，也可以基于生成器来实现，不过需要借助python 标准库提供的contextlib模块，contextlib模块提供了contextmanager装饰器来帮我们实现。

from contextlib import contextmanager

@contextmanager
def MyFile(filepath,mode):
  f = open(filepath,mode=mode,encoding='utf-8')
  yield f
  print('------')
  f.close()

with MyFile('1.txt','w') as myf:
  myf.write('123\n')

------

　　以上代码执行顺序为：

　　1、执行MyFile()方法，执行函数内第一行打开文件

　　2、执行yield f，MyFile方法返回，f会赋值给with语句块的myf变量

　　3、执行with语句块内的逻辑，往文件内写数据

　　4、回到MyFile方法中，执行yield后面的代码关闭文件

　　在使用contextmanager装饰器时，如果被装饰的方法内发生了异常，则将不会执行yield之后的逻辑，所以我们需要在自己的方法中进行异常处理。

from contextlib import contextmanager

@contextmanager
def MyFile(filepath,mode):
  f = open(filepath,mode=mode,encoding='utf-8')
  yield f
  print(haha)
  print('------') # 此处代码未执行
  f.close()

with MyFile('1.txt','w') as myf:
  myf.write('123\n')

Traceback (most recent call last):
  File "C:\Users\Administrator\Desktop\test.py", line 70, in <module>
    myf.write('123\n')
  File "D:\install\python3\lib\contextlib.py", line 119, in __exit__
    next(self.gen)
  File "C:\Users\Administrator\Desktop\test.py", line 65, in MyFile
    print(haha)
NameError: name 'haha' is not defined

　　在自己的方法中加入异常处理。

from contextlib import contextmanager

@contextmanager
def MyFile(filepath,mode):
  try:
    f = open(filepath,mode=mode,encoding='utf-8')
    yield f
    print(haha)
  finally:
    print('------') # 此处代码未执行
    f.close()

with MyFile('1.txt','w') as myf:
  myf.write('123\n')

很赞哦！

python编程网提示：转载请注明来源www.python66.com。
有宝贵意见可添加站长微信(底部)，获取技术资料请到公众号(底部)。同行交流请加群