Python上下文管理器学习

   上下文管理器(context
manager)用于规定某个对象的使用范围。进入或者离开该使用范围，会有特殊操作被调用。它的语法形式是with...as...

我们经常会进行这样的操作：打开文件，读写，关闭文件。偶尔会忘记关闭文件，使用上下文管理器可以在不需要文件的时候，自动关闭文件。



1 with ...as...示例

with open("test.txt", "w") as file:
print(file.closed)
file.write("with as test!")
print(file.closed)
   使用上下文管理器，当隶属的程序块执行结束的时候(也就是不再缩进)，上下文管理器自动关闭了文件
(file.closed查询文件是否关闭)。使用缩进规定了文件对象f的使用范围。
 
 当使用上下文管理器时，Python在进入程序块之前调用对象的__enter__()方法，在结束程序块的时候调用__exit__()方法。对于文件对象file来说，它定义了__enter__()和__exit__()方法(通过dir(file)可查看)。在file的__exit__()方法中，调用了self.close()语句。

2
自定义上下文管理器

 
所谓自定义上下文管理器，其实就是自己定义__enter__()和__exit__()方法。我们需要创建一个类，然后在其中定义__enter__()和__exit__()。

class test:
def __init__(self, input):
self.text = input
def __enter__(self):
self.text = "call __enter__ " + self.text
return self #return an object
def __exit__(self,exc_type,exc_value,traceback):
self.text = self.text + " call __exit__"

with test("hello") as mytest:
print(mytest.text)
print dir(mytest)
print(mytest.text)

我们注意到: __exit__()有四个参数。当程序块中出现异常，__exit__()的参数中exc_type, exc_value, traceback用于描述异常。我们可以根据这三个参数进行相应的处理。如果正常运行结束，这三个参数都是None。