Python上下文管理器 context manager（with...as...）

一、概念

上下文管理器：就是实现了上下文管理协议的对象。主要用于保存和恢复各种全局状态，关闭文件等。上下文管理器本身是一种装饰器。

上下文允许可以自动的开始和结束一些和事情。例如当利用with...as打开一个文件时，python就自动创建了一个上下文管理器。

 

二、上下文管理协议

上下文管理协议包含两个方法：

contextmanager.__enter__(）：with语句中的代码块执行前，会执行__enter__()，返回的值将赋值给with句中的as后的变量。

contextmanager.__exit__()：with语句中的代码块执行结束或者出错，会执行__exit__()。

 

三、实例 with...as 代替try...except...finally打开文件

# 错误的写法
try:
    f = open("xxx")
    do something
except:
    do something
finally:
    f.close

##################

# 正确的写法
'''要写finally，是因为防止程序抛出异常最后不能关闭文件，
但是需要关闭文件有一个前提就是文件已经打开了。 在第一段错误代码中，
如果异常发生在f=open(‘xxx’)的时候，比如文件不存在，
立马就可以知道执行f.close()是没有意义的，改正后的解决方案就是第二段代码'''

try：
    f = open("xxx")
except:
    print("fail to open")
    exit(-1)
try:
    do something
except:
    do something
finally:
    f.close()

with...as...:

#打开一个文件并保证最后关闭它
with open("xxx") as f:
    data = f.read()
    do something with data

# 打开多个
with open("xxx") as f1, open("yyy") as f2:
   do something with f1, f2

四、自定义上下文管理器

class MyContextManager(object):
    def __init__(self):
        print("init 函数被调用")

    def __enter__(self):
        print("enter函数被调用，返回一个对象")
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("exit 函数被调用")

with MyContextManager():
    print("创建实例做点什么")

################
#结果：
init 函数被调用
enter函数被调用，返回一个对象
创建实例做点什么
exit 函数被调用