上下文管理器:任何实现了__enter__()和__exit__()方法的对象都可以称之为上下文管理器上下文管理器对象可以使用with关键字,显然文件(file)对象也实现了上下文管理器 。
# with与上下文管理器:
# 正确关闭一个文件 1.普通版
def m1():
f = open("output.txt", "w")
f.write("python 蟒蛇")
f.close()
# 这样会有一个潜在的问题,如果在调用write过程中,出现了异常进而导致后续代码无法继续执行
# close()方法无法被正常调用,因此资源就会一直被该程序占用
# 2 进阶版
def m2():
f = open("output.txt", "w")
try:
f.write("python 蟒蛇")
except Exception as e:
print(str(e))
finally:
f.close()
# 改良版本的程序是对可能发生异常的代码处进行捕获,该语句表示
# 如果在try代码块中程序出现了异常,后续代码就不在执行而是直接跳转到except
# 代码块,但是无论如何finally中的代码最终都会被执行,因此将close()
# 方法放在finally代码中文件就一定会关闭
# 3.高级版:
def m3():
with open("out.txt", "w") as f:
f.write("python 蟒蛇")
# open方法的返回值赋给变量f,当离开with代码块时,系统会自动调用
# f.close()方法,那么with的实现的原理是什么呢?
# 上下文管理器
# 任何实现了__enter__()和__exit__()方法的对象都可以称之为上下文管理器
# 上下文管理器对象可以使用with关键字,显然文件(file)对象也实现了上下文管理器
# 1.下面我们自己模拟实现一个自己的文件类,让该类实现__enter__()
# 和__exit__()方法
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方法,也会被系统自动的去调用
# 2. 实现上下文管理器的另外方式
# python中还提供了一个contextmanger的装饰器,进一步简化了上下文管理器的实现方式,通过yield将函数分割成两部分,yield之前的语句在__enter__()
# 方法中执行,yield之后的语句在__exit__()方法中执行,紧跟在yield后面的值是函数的返回值
@contextmanager
def my_open(path, mode):
f = open(path, mode)
yield f
f.close()
# 调用
with my_open("out.put", "w") as f:
f.write("hello,the simplest context manager")
#总结:with语法用于简化资源操作的后续操作,是try/finally的替代方法
#实现原理建立在上下文管理器上