Skip to content

Commit 2514a63

Browse files
authored
bpo-29587: Enable implicit exception chaining with gen.throw() (GH-19811)
Before this commit, if an exception was active inside a generator when calling gen.throw(), then that exception was lost (i.e. there was no implicit exception chaining). This commit fixes that.
1 parent c001c09 commit 2514a63

File tree

3 files changed

+24
-0
lines changed

3 files changed

+24
-0
lines changed

Lib/test/test_generators.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -316,6 +316,23 @@ def g():
316316
self.assertEqual(cm.exception.value.value, 2)
317317

318318

319+
class GeneratorThrowTest(unittest.TestCase):
320+
321+
def test_exception_context_set(self):
322+
def f():
323+
try:
324+
raise KeyError('a')
325+
except Exception:
326+
yield
327+
328+
gen = f()
329+
gen.send(None)
330+
with self.assertRaises(ValueError) as cm:
331+
gen.throw(ValueError)
332+
context = cm.exception.__context__
333+
self.assertEqual((type(context), context.args), (KeyError, ('a',)))
334+
335+
319336
class YieldFromTests(unittest.TestCase):
320337
def test_generator_gi_yieldfrom(self):
321338
def a():
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Enable implicit exception chaining when calling :meth:`generator.throw`.

Objects/genobject.c

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -512,6 +512,12 @@ _gen_throw(PyGenObject *gen, int close_on_genexit,
512512
}
513513

514514
PyErr_Restore(typ, val, tb);
515+
if (gen->gi_exc_state.exc_type) {
516+
Py_INCREF(gen->gi_exc_state.exc_type);
517+
Py_XINCREF(gen->gi_exc_state.exc_value);
518+
Py_XINCREF(gen->gi_exc_state.exc_traceback);
519+
_PyErr_ChainExceptions(gen->gi_exc_state.exc_type, gen->gi_exc_state.exc_value, gen->gi_exc_state.exc_traceback);
520+
}
515521
return gen_send_ex(gen, Py_None, 1, 0);
516522

517523
failed_throw:

0 commit comments

Comments
 (0)