python 上下文管理器 with 语句

上下文管理器 with 语句可以自动地分配和释放资源。

读写文件

这篇文章中介绍了文件读写的方法，每次读取文件时要使用 open() 函数打开文件，使用结束后要用 close() 函数关闭文件。有时，我们会忘记关闭文件，或者打开了许多文件占用太多资源。使用上下文管理器 with 语句可以方便地读取文件，它会自动地释放文件。

with open('test.txt') as f:
	f.read()

with 语句后可进行文件操作，它会自动关闭文件，而不用去调用 close() 函数。

基于类的上下文管理器

用类创建上下文管理器，需要包括 __enter__() 方法，返回需要被管理地资源，也要包括 __exit__() 方法，保存一些释放、清理资源的操作。

class Manager():
	def __init__(self):
		pass
	def __enter__(self):
		pass
	def __exit__(self, exc_type, exc_val, exc_tb):
		pass

__exit__() 方法中的参数分别表示 exception_type、exception_value 和 traceback，当执行 with 语句时，如果有异常抛出，异常的信息就会包含在这三个变量中，传入方法 __exit__()，这样就可以在该方法中处理异常了。如果确定异常已经被处理了，请在 __exit__() 的最后，加上 return True 这条语句，否则异常仍然会被抛出。

with Manager() as f:
	pass

当使用 with 语句创建上下文管理器时，首先会调用 __init__() 函数创建对象；其次调用 __enter__() 函数，返回的 Manager 对象赋值给变量 f，再进行 with 下的操作；最后调用 __exit__() 语句释放资源。

基于生成器的上下文管理器

@contextmanager
def manager(name, mode):
	try:
		f = open(name, mode)
		yield f
	finally:
		f.close()

with manager('test.txt', 'w') as f:
	pass

这段代码中 manager 即是一个生成器，with 语句会打开文件，返回文件对象 f，结束后关闭文件。务必加上装饰器 @contextmanager。