如何让返回生成器的上下文管理器在未完全消费时正常退出？

方法1：用生成器包裹内置open的上下文 #

最简洁的方式是把文件读取的上下文逻辑直接封装在生成器函数内部，利用Python生成器的中断清理机制，保证即使提前终止遍历，文件也会被正常关闭。

示例代码：

import csv

class Model:
    def __init__(self, *fields):
        self.fields = fields

def create_models(csv_path):
    # 把文件打开的with块放在生成器内部
    with open(csv_path, 'r', newline='') as csv_file:
        reader = csv.reader(csv_file)
        next(reader)  # 跳过表头（如果需要）
        for row in reader:
            yield Model(*row)

# 使用示例
for model in create_models('data.csv'):
    print(model.fields)
    if model.fields[0] == '5':  # 提前终止条件
        break

原理 #

当你在遍历生成器时提前break，Python会向生成器抛出GeneratorExit异常，生成器函数会进入清理流程——内部的with块会自动执行__exit__方法，关闭文件，完全符合你的需求，而且不需要修改任何内置上下文管理器。

方法2：自定义上下文管理器类 #

如果需要更明确的资源控制逻辑，可以写一个自定义上下文管理器，将生成器作为类的方法，确保__exit__方法无论遍历是否完成都会被调用。

示例代码：

import csv

class Model:
    def __init__(self, *fields):
        self.fields = fields

class CSVModelReader:
    def __init__(self, csv_path):
        self.csv_path = csv_path
        self.csv_file = None
        self.reader = None

    def __enter__(self):
        # 在__enter__中打开文件
        self.csv_file = open(self.csv_path, 'r', newline='')
        self.reader = csv.reader(self.csv_file)
        next(self.reader)  # 跳过表头
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 无论是否提前终止，都会执行文件关闭
        if self.csv_file:
            self.csv_file.close()
        # 返回False表示不抑制任何异常，让上层处理
        return False

    def create_models(self):
        # 生成器方法按需生成Model实例
        for row in self.reader:
            yield Model(*row)

# 使用示例
with CSVModelReader('data.csv') as reader:
    for model in reader.create_models():
        print(model.fields)
        if model.fields[0] == '5':
            break

原理 #

with语句的核心机制是：无论块内代码是正常结束、break还是抛出异常，都会调用上下文管理器的__exit__方法。这里生成器只是按需提供数据，资源的生命周期完全由上下文管理器掌控，所以提前终止遍历不会影响文件的关闭。

这两种方案都是同步实现，不需要引入async/await，完全符合你的需求。

内容的提问来源于stack exchange，提问作者ezgrs