Python上下文管理器如何实现
今天小编给大家分享一下Python上下文管理器如何实现的相关知识点,内容详细,逻辑清晰,相信大部分人都还太了解这方面的知识,所以分享这篇文章给大家参考一下,希望大家阅读完这篇文章后有所收获,下面我们一起来了解一下吧。
创新互联主营盈江网站建设的网络公司,主营网站建设方案,重庆APP开发,盈江h5重庆小程序开发搭建,盈江网站营销推广欢迎盈江等地区企业咨询
什么时候可以考虑上下文管理器
当你的代码逻辑需要用到如下关键字时,可以考虑使用上下文管理器让你的代码更加优雅:
try: ... finally: ...
接下来介绍实现上下文管理器的三种方法。
方法1(上下文管理器协议)
总所周知,open()是默认支持上下文管理器的。所以打开一个txt文件,并向里面写入内容,再关闭这个文件的代码可以这样写:
with open("1.txt", "w") as file: file.write("this is a demo")
这是等同于:
file = None try: file = open("1.txt", "w") file.write("this is a demo") finally: file.close()
要在Python中实现with语句的使用,就需要借助上下文管理器协议。也就是需要实现__enter__和__exit__两个魔法方法。
class OpenMyFile(object): def __init__(self, path): self.path = path def __enter__(self): print("opening the txt") self.f = open(self.path, "w") return self def __exit__(self, *args, **kwargs): print("closing the txt") self.f.close() def write(self, string): print("writing...") self.f.write(string) with OpenMyFile("2.txt") as file: file.write("this is a demo2") # 输出: opening the txt writing... closing the txt
同时能够看到本地生成了2.txt文件。需要注意的是,__enter__得return实例对象,不然会报异常:AttributeError: "NoneType" object has no attribute "write"
这是因为Python中的函数默认返回None。
方法2(@contextmanager)
利用contextlib中的contextmanager装饰器。
from contextlib import contextmanager @contextmanager def open_my_file(path): print("opening the txt") f = open("3.txt", "w") yield f print("closing the txt") f.close() with open_my_file("3.txt") as file: file.write("this is demo3") # 输出: opening the txt closing the txt
在@contextmanager装饰的函数中,需要用yield隔开两个逻辑语句。这里yield出来的对象会被as后面的变量接收。
方法3(contextlib.closing())
利用contextlib中的closing()方法。
from contextlib import closing class OpenMyFile(object): def __init__(self, path): print("opening the txt") self.f = open(path, "w") def write(self, string): self.f.write(string) def close(self): print("closing the txt") self.f.close() with closing(OpenMyFile("4.txt")) as file: file.write("this is demo4") # 输出: opening the txt closing the txt
与方法1不同。经过closing()方法包装过后,在with语句结束时,会强制调用对象的close()方法。所以使用方法3时,需要定义的方法不是__exit__()而是close()。
以上就是“Python上下文管理器如何实现”这篇文章的所有内容,感谢各位的阅读!相信大家阅读完这篇文章都有很大的收获,小编每天都会为大家更新不同的知识,如果还想学习更多的知识,请关注创新互联行业资讯频道。
标题名称:Python上下文管理器如何实现
转载来于:http://scyanting.com/article/jigphe.html