Python 上下文管理器

什么是上下文管理器？

这里先了解下上下文，上下文是 $context$ 直译的叫法，在程序中用来表示代码执行过程中所处的前后环境，比如在文件操作时，文件需要打开关闭，而文件读写操作就处于文件操作的上下文环境中

上下文管理器是指在一段代码执行之前，执行一些预处理的工作，代码执行之后再执行一些清理工作

上下文管理器中有__enter__()和 __exit__() 两个方法，__enter__()方法在执行 with 后面的语句时执行，一般用来处理操作前的内容,比如一些创建对象，初始化等； __exit__() 方法在 with 内的代码执行完毕后执行，一般用来处理一些善后收尾工作，比如文件的关闭，数据库的关闭等

简单来说，上下文管理器的原理过程如下：

• 调用__enter__()方法，进行预处理操作
• 执行用户操作
• 调用 __exit__() 方法，完成清理操作

上下文管理器的应用场景：

资源管理功能，即文件处理、网络连接、数据库连接等操作时需要关闭资源

也可以在代码执行前后增加功能，类似于装饰器，比如代码之前做权限验证等

上下文管理器的使用

例子 $1$

# 创建一个文件写入字符串“Python”
f = open('123.txt', 'w')
f.write("python")
f.close()

# 使用with语句调用上下文实现文件写入操作
with open('123.txt', 'w') as f:
    f.write('python')

例子 $2$

#自定义一个上下文管理类
class MyOpen():
    def __init__(self):
        print("初始化方法")
    def __enter__(self):
        print("执行enter方法")
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("执行exit方法")
print("实例化对象")
t = MyOpen()
print("with语句")
with MyOpen() as f:
    print("执行代码")

输出：
实例化对象
初始化方法
with语句
初始化方法
执行enter方法
执行代码
执行exit方法

例子 $3$

import contextlib
import time

@contextlib.contextmanager
def timeit(title):
    print('第一行')
    start = time.time()
    yield
    print('第二行')
    end = time.time()
    usedTime = (end - start) * 1000
    print('Use time %d ms' % usedTime)

with timeit(1):
    print('第三行')
    time.sleep(1)

with timeit(2):
    print('第四行')
    time.sleep(2)
    
输出：
第一行
第三行
第二行
Use time 1001 ms
第一行
第四行
第二行
Use time 2008 ms

参考

• https://zhuanlan.zhihu.com/p/330610676
• https://blog.csdn.net/l1l1l1l/article/details/92702773