with上下文管理器

上下文管理器

任何实现了 enter() 和 exit() ⽅法的对象都可称之为上下⽂管理
器，上下⽂管理器对象可以使⽤ with 关键字。显然，⽂件（file）对象也实现
了上下⽂管理器

方法一

class File():
	def __init__(self, filename, mode):
		self.filename = filename
		self.mode = mode
	def __enter__(self):
		print("entering")
		self.f = open(self.filename, self.mode)
		return self.f
	def __exit__(self, *args):
		print("will exit")
		self.f.close()

enter() ⽅法返回资源对象，这⾥就是你将要打开的那个⽂件对象，
exit() ⽅法处理⼀些清除⼯作。
因为 File 类实现了上下⽂管理器，现在就可以使⽤ with 语句了。

with File('out.txt', 'w') as f:
	print("writing")
	f.write('hello, python')

这样，你就⽆需显示地调⽤ close ⽅法了，由系统⾃动去调⽤，哪怕中间遇
到异常 close ⽅法也会被调⽤。

方法二

定义

from contextlib import contextmanager
@contextmanager
def my_open(path, mode):
	f = open(path, mode)
	yield f
	f.close()

调⽤

with my_open('out.txt', 'w') as f:
	f.write("hello , the simplest context manager")