Python深入学习之上下文管理器
上下文管理器(contextmanager)是Python2.5开始支持的一种语法,用于规定某个对象的使用范围。一旦进入或者离开该使用范围,会有特殊操作被调用(比如为对象分配或者释放内存)。它的语法形式是with...as...
关闭文件
我们会进行这样的操作:打开文件,读写,关闭文件。程序员经常会忘记关闭文件。上下文管理器可以在不需要文件的时候,自动关闭文件。
下面我们看一下两段程序:
#withoutcontextmanager f=open("new.txt","w") print(f.closed) #whetherthefileisopen f.write("HelloWorld!") f.close() print(f.closed)
以及:
#withcontextmanager withopen("new.txt","w")asf: print(f.closed) f.write("HelloWorld!") print(f.closed)
两段程序实际上执行的是相同的操作。我们的第二段程序就使用了上下文管理器(with...as...)。上下文管理器有隶属于它的程序块。当隶属的程序块执行结束的时候(也就是不再缩进),上下文管理器自动关闭了文件(我们通过f.closed来查询文件是否关闭)。我们相当于使用缩进规定了文件对象f的使用范围。
上面的上下文管理器基于f对象的__exit__()特殊方法(还记得我们如何利用特殊方法来实现各种语法?参看特殊方法与多范式)。当我们使用上下文管理器的语法时,我们实际上要求Python在进入程序块之前调用对象的__enter__()方法,在结束程序块的时候调用__exit__()方法。对于文件对象f来说,它定义了__enter__()和__exit__()方法(可以通过dir(f)看到)。在f的__exit__()方法中,有self.close()语句。所以在使用上下文管理器时,我们就不用明文关闭f文件了。
自定义
任何定义了__enter__()和__exit__()方法的对象都可以用于上下文管理器。文件对象f是内置对象,所以f自动带有这两个特殊方法,不需要自定义。
下面,我们自定义用于上下文管理器的对象,就是下面的myvow:
#customizedobject
classVOW(object): def__init__(self,text): self.text=text def__enter__(self): self.text="Isay:"+self.text #addprefix returnself #note:returnanobject def__exit__(self,exc_type,exc_value,traceback): self.text=self.text+"!" #addsuffix
withVOW("I'mfine")asmyvow: print(myvow.text)
print(myvow.text)