1、with上下文管理器的用法
- with用的最多的可能就是打开文件,读写文件的场景,如下代码:
with open("demo.txt","w+",encoding="utf8") as f:
f.write("hello world")
其等同于如下代码:
f=open("demo.txt","w+",encoding="utf8")
f.write("hello world")
f.close()
使用with语句的时候不需要显式的去关闭文件资源,因为它自动会关闭,那么这个自动关闭是怎么实现的呢,这其实就是__enter__和__exit__魔法函数在起作用
2、__enter__和__exit__魔法函数的工作原理
- 以上面with as 的形式,那么首先是进入类函数中__enter__魔法函数,__enter__魔法函数返回结果赋值给as后面的变量,当with as 语句结束时,会自动调用__exit__魔法函数
下面重写写一个类似open的类,来实现with as 打开文件读写文件的实例,如下:
class OpenFile(object):
def __init__(self,file_name="",mode="r",encoding="utf8",**kwargs):
print("-----------------in init------------------")
self.__file=open(file_name,mode,encoding=encoding)
def __enter__(self):
print("-----------------in enter ----------------")
return self
def write(self,ctx):
print("-----------------begin write-------------")
self.__file.write(ctx)
print("-----------------finish write-------------")
def __exit__(self,exec_type,exec_value,exec_tb):
print("-----------------in exit------------------")
self.__file.close()
with OpenFile("demo.txt","w+",encoding="utf8") as f:
f.write("hello world")
执行结果如下:
-----------------in init------------------
-----------------in enter ----------------
-----------------begin write-------------
-----------------finish write-------------
-----------------in exit------------------
因此,如果想实现一个自定义的上下文管理器,只需要在自定义类中实现这两个魔法函数即可,这里有一个关键点需要理解的就是with as 语句中as 后面的变量就是__enter__魔法函数返回值,理解了这一点,就很容易实现自己的上下文管理器了