]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
[3.13] gh-142766: Clear frame when `generator.close()` is called (gh-142838) (#142900)
authorSam Gross <colesbury@gmail.com>
Wed, 17 Dec 2025 18:46:23 +0000 (13:46 -0500)
committerGitHub <noreply@github.com>
Wed, 17 Dec 2025 18:46:23 +0000 (18:46 +0000)
(cherry picked from commit 25397f9541be89264d35d41a67defcfeaa950844)

Lib/test/test_generators.py
Misc/NEWS.d/next/Core and Builtins/2025-12-16-11-56-20.gh-issue-142766.Uy2HTm.rst [new file with mode: 0644]
Objects/genobject.c

index e48d79d34f479f5ad34a366f5bf24ab9e37b5a28..d6014fd9d63cdf269188ceffed8a58b7d618c2d2 100644 (file)
@@ -246,6 +246,33 @@ class GeneratorTest(unittest.TestCase):
         #This should not raise
         loop()
 
+    def test_close_clears_frame(self):
+        # gh-142766: Test that closing a generator clears its frame
+        class DetectDelete:
+            def __init__(self):
+                DetectDelete.deleted = False
+
+            def __del__(self):
+                DetectDelete.deleted = True
+
+        def generator(arg):
+            yield
+
+        # Test a freshly created generator (not suspended)
+        g = generator(DetectDelete())
+        g.close()
+        self.assertTrue(DetectDelete.deleted)
+
+        # Test a suspended generator
+        g = generator(DetectDelete())
+        next(g)
+        g.close()
+        self.assertTrue(DetectDelete.deleted)
+
+        # Clear via gi_frame.clear()
+        g = generator(DetectDelete())
+        g.gi_frame.clear()
+        self.assertTrue(DetectDelete.deleted)
 
 class ModifyUnderlyingIterableTest(unittest.TestCase):
     iterables = [
diff --git a/Misc/NEWS.d/next/Core and Builtins/2025-12-16-11-56-20.gh-issue-142766.Uy2HTm.rst b/Misc/NEWS.d/next/Core and Builtins/2025-12-16-11-56-20.gh-issue-142766.Uy2HTm.rst
new file mode 100644 (file)
index 0000000..6a14976
--- /dev/null
@@ -0,0 +1 @@
+Clear the frame of a generator when :meth:`generator.close` is called.
index 0273dea958efc11cd46f02583f5ad1a8fe23029a..412d3c1090b5229edfda96328a37265879c6bdf8 100644 (file)
@@ -366,6 +366,7 @@ gen_close(PyGenObject *gen, PyObject *args)
 
     if (gen->gi_frame_state == FRAME_CREATED) {
         gen->gi_frame_state = FRAME_COMPLETED;
+        gen_clear_frame(gen);
         Py_RETURN_NONE;
     }
     if (FRAME_STATE_FINISHED(gen->gi_frame_state)) {