Python基础教程：上下文管理器 context manager（with...as...）

╰+攻爆jí腚メ 2022-10-13 14:52 152阅读 0赞

**一、概念**

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

上下文允许可以自动的开始和结束一些和事情。例如当利用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...`:

''' 遇到问题没人解答？小编创建了一个Python学习交流QQ群：531509025 寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！ '''
    #打开一个文件并保证最后关闭它
    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 函数被调用