Python 的 with 语句、上下文管理协议

file = open('some_file.txt', 'w')
try:
    file.write('Hello, World!')
finally:
    file.close() # 确保在任何情况下文件都会被关闭

with open('some_file.txt', 'w') as file:
    file.write('Hello, World!')
# 一旦离开 with 代码块，文件会自动关闭，无需手动调用 file.close()

创建上下文管理器

使用类的方法

import time

class Timer:
    def __enter__(self):
        self.start_time = time.time()
        print("开始计时...")
        return self.start_time # 可以返回自身或其他对象，供 with...as 使用

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.end_time = time.time()
        elapsed_time = self.end_time - self.start_time
        print(f"代码块运行耗时: {elapsed_time:.4f} 秒")
        # 返回 False，让任何发生的异常正常抛出
        return False

# 使用自定义的上下文管理器
with Timer() as t:
    print(f"开始时间：{t}")
    # 模拟一些耗时操作
    time.sleep(2)
    # 如果这里发生异常，计时功能依然会正常工作
    # print(1 / 0) # 取消注释测试异常情况

# 输出：
# 开始计时...
# 开始时间：1755850271.0199914
# 代码块运行耗时: 2.0002 秒

使用contextmanager

利用生成器函数（带yield语句的函数）

from contextlib import contextmanager
import time

@contextmanager
def timer():
    start_time = time.time()
    print("开始计时...")
    try:
        yield start_time
        # 在这里 yield 之前的部分相当于 __enter__
    finally:
        # finally 块确保了清理工作一定会执行，相当于 __exit__
        end_time = time.time()
        elapsed_time = end_time - start_time
        print(f"代码块运行耗时: {elapsed_time:.4f} 秒")

# 使用方式完全相同
with timer() as t:
    print(f"开始时间：{t}")
    time.sleep(1.5)

使用装饰器定义上下文管理器则要使用@asynccontextmanager 

定义相关函数使用async，调用相关函数使用await。