contextlib 实现效果和with 语句一样,他有他独特的语法来实现 with 的enter 和exit方法. 请看下边的例子.
import contextlib
@contextlib.contextmanager # contextlib装饰器 可以将函数变成 一个上下文管理器
def file_open(file_name):
print("file name") # 在yield 之前相当于 with 的 enter函数
yield {} #这个标识就是生成器,生成器的内容我将会专门写.
print("file end") #在yield 之后相当于 with 的 exit函数
#在适用的时候,直接用with调用.
with file_open("test.txt") as fp:
print("yield body") #这个才是我们要运行的主题部分
打印结果:
file name
yield body
file end
封装好的上下文生成器, 会先运行yield 之前的代码,进行准备工作,例如获取资源等,
再运行主体部分
主体结束后 ,就开始进行清理工作,就和上一节讲的exit 函数一样.
为什么会实现这样的效果,利用 生成器的特性. 自己可以先查下,以后我也会讲述.