contextlib简化上下文管理器

版权声明:转载 或者复制请标注来源 https://blog.csdn.net/qq_34979346/article/details/83270526

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 函数一样.
为什么会实现这样的效果,利用 生成器的特性. 自己可以先查下,以后我也会讲述.

猜你喜欢

转载自blog.csdn.net/qq_34979346/article/details/83270526
今日推荐