python генераторы сборка мусора

Вы, как и ожидали, полагаетесь на специфичное для реализации поведение подсчета ссылок CPython.¹

На самом деле, если вы запустите этот код, скажем, в PyPy, вывод обычно будет:

Counting...  10
Counting...  9
Counting...  8
Counting...  7
Counting...  6
Finished counting
In the finally block

И если вы запустите его в интерактивном сеансе PyPy, эта последняя строка может появиться много строк позже или даже только тогда, когда Вы наконец выйдете.

---

Если вы посмотрите, как реализуются генераторы, у них есть методы примерно такие:

def __del__(self):
    self.close()
def close(self):
    try:
        self.raise(GeneratorExit)
    except GeneratorExit:
        pass

CPython удаляет объекты сразу же, когда счетчик ссылок становится нулевым (он также имеет сборщик мусора для разбиения циклических ссылок, но это не имеет значения здесь). Как только генератор выходит из области действия, он удаляется, поэтому он закрывается, поэтому он поднимает GeneratorExit в рамку генератора и возобновляет его. И, конечно, нет никакого обработчика для GeneratorExit, поэтому предложение finally выполняется, и управление передается вверх по стеку, где исключение проглатывается.

В PyPy, который использует гибридный мусор коллектор, генератор не будет удален до следующего раза, когда GC решит сканировать. И в интерактивном сеансе, с низким давлением памяти, это может быть так же поздно, как время выхода. Но как только это происходит, происходит то же самое.

Вы можете увидеть это, обработав GeneratorExit явно:

def countdown(n):
    try:
        while n > 0:
            yield n
            n -= 1
    except GeneratorExit:
        print('Exit!')
        raise
    finally:
        print('In the finally block')

(Если вы оставите raise выключенным, вы получите те же результаты только по несколько иным причинам.)

---

Вы можете явно close генератор-и, в отличие от материала выше, это часть открытого интерфейса генератора типа:

def main():
    c = countdown(10)
    for n in c:
        if n == 5:
            break
        print('Counting... ', n)
    c.close()
    print('Finished counting')

Или, Конечно, вы можете использовать оператор with:

def main():
    with contextlib.closing(countdown(10)) as c:
        for n in c:
            if n == 5:
                break
            print('Counting... ', n)
    print('Finished counting')

---

_{1. Как указывает ответТима Питерса , вытакже полагаетесь на специфичное для реализации поведение компилятора CPython во втором тесте.}