关于with上下文管理器的一些小总结:
没有使用with打开文件的时候:
def m1():
f = open("output.txt", "w")
f.write("python之禅")
f.close()
这样写有一个潜在的问题,如果在调用 write 的过程中,出现了异常进而导致后续代码无法继续执行,close 方法无法被正常调用,因此资源就会一直被该程序占用者释放。
使用with打开文件:
def m3():
with open("output.txt", "r") as f:
f.write("Python之禅")
可以看到 我们的代码量明显减少,而且使用with有一个明显的优点:当我们打开一个文件进行操作的时候,不用手动关闭文件,with会自动调用open里面的上下文管理器进行关闭,当打开文件进行操作的时候,有误操作报错的时候,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 方法也会被调用。
另外,with 不仅可以用在打开文件上,所有打开操作,只要里面包含__enter __ 和 __ exit__方法,都可以使用with触发
总结:
with 是用来触发上下文管理器的方法,我们使用这个方法可以避免打开文件进行操作时,误操作出现错误无法关闭文件造成的资源浪费,with不仅可以使用在打开文件上,所有打开操作,内部只要存在__enter __ 和 __ exit__方法,都可以使用with触发