如何创建可被正确垃圾回收的Python自定义生成器类？

你观察得非常准确——继承collections.abc.Generator确实只让你的类满足了生成器的接口规范，但CPython中只有原生的GeneratorType实例才会在垃圾回收时自动触发close()方法，这是解释器层面的专属逻辑，ABC本身并没有实现这个行为。

为什么会这样？ #

collections.abc.Generator只是一个抽象基类，它定义了生成器必须实现的send()、throw()、close()方法，但没有涉及垃圾回收时的自动处理逻辑。而原生生成器（用yield定义的函数返回的对象）属于types.GeneratorType，CPython在底层为这个类型做了特殊处理：当实例被垃圾回收时，会自动调用其close()方法，进而抛出GeneratorExit异常让生成器内部执行清理。

你没法让自定义类成为GeneratorType的实例——这个类型是CPython内部实现的，不允许用户继承或创建其实例（尝试继承会直接报错）。

如何解决：手动实现自动清理逻辑 #

要让你的自定义生成器类在垃圾回收时自动执行清理，你需要自己添加__del__方法，在其中调用close()。同时，要确保close()方法能正确处理资源释放，并且和原生生成器的行为对齐。

修改后的代码示例：

from collections.abc import Generator
from threading import Lock
import time
lock = Lock()

class CustomGeneratorClass(Generator):
    def __init__(self, n, lock):
        super().__init__()
        self.lock = lock
        self.lock.acquire()
        print("Generator Class Initialised: I grabbed a lock")
        self.n = n
        self.c = 0
        self.closed = False  # 标记是否已关闭，避免重复执行清理

    def send(self, arg):
        if self.closed:
            raise ValueError("Generator is closed")
        value = self.c
        if value >= self.n:
            self.close()
            raise StopIteration
        self.c += 1
        return value

    def throw(self, typ, value=None, traceback=None):
        if typ is GeneratorExit:
            # 捕获GeneratorExit时调用close()，和原生生成器行为一致
            self.close()
            raise
        # 其他异常交给父类处理（或者自定义逻辑）
        super().throw(typ, value, traceback)

    def close(self):
        if not self.closed:
            print("Generator Class Closed: I let go of the lock")
            self.lock.release()
            self.closed = True

    def __del__(self):
        # 垃圾回收时自动调用close()
        self.close()

# 测试代码
for i in CustomGeneratorClass(100, lock):
    print("Received ", i)
    time.sleep(0.02)
    if i == 3:
        break

if not lock.acquire(blocking=False):
    print("Oops: Finished, but lock wasn't released")
else:
    print("Finished: Lock was free")
    lock.release()

关键细节说明： #

1. __del__方法：当对象被垃圾回收时，Python会调用这个方法，我们在这里触发close()来执行清理。注意__del__有一些限制（比如循环引用可能导致对象无法被及时回收），但对于生成器这种通常不会有循环引用的场景，这个方法非常有效。
2. close()方法：添加了closed标记，避免重复执行资源释放逻辑（比如多次调用lock.release()会报错）。
3. throw()方法：处理GeneratorExit异常时调用close()，对齐原生生成器的行为——原生生成器的close()方法本质就是调用throw(GeneratorExit)。
4. send()方法：检查生成器是否已关闭，避免在关闭后继续调用导致错误。

运行修改后的代码，你会看到预期的输出：

Generator Class Initialised: I grabbed a lock
Received  0
Received  1
Received  2
Received  3
Generator Class Closed: I let go of the lock
Finished: Lock was free

总结 #

• collections.abc.Generator只是接口规范，不包含垃圾回收时自动调用close()的逻辑
• 原生GeneratorType的自动close()是CPython底层的特殊处理，用户无法复用这个逻辑到自定义类
• 要实现自动清理，必须在自定义类中添加__del__方法，手动调用close()，同时确保close()、throw()方法的行为和原生生成器对齐

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