bpo-32810: Allow accessing the ags_gen and agt_gen attrs of async generators

ZackerySpytz · ZackerySpytz · commit 6024d01ecc12 · 2018-12-14T12:53:32.000-07:00
diff --git a/Lib/test/test_asyncgen.py b/Lib/test/test_asyncgen.py
@@ -374,6 +374,15 @@ async def gen():
 
         self.assertTrue(inspect.isawaitable(g.aclose()))
 
+    def test_async_gen_ags_gen_agt_gen(self):
+        async def agen():
+            yield 1
+        g = agen()
+        asend = g.asend(1)
+        self.assertIsInstance(asend.ags_gen, types.AsyncGeneratorType)
+        athrow = g.athrow()
+        self.assertIsInstance(athrow.agt_gen, types.AsyncGeneratorType)
+
 
 class AsyncGenAsyncioTest(unittest.TestCase):
 
diff --git a/Misc/NEWS.d/next/Library/2018-12-14-12-53-04.bpo-32810.A2g1Cp.rst b/Misc/NEWS.d/next/Library/2018-12-14-12-53-04.bpo-32810.A2g1Cp.rst
@@ -0,0 +1,2 @@
+Allow accessing the ``ags_gen`` and ``agt_gen`` attributes of asynchronous
+generators.
diff --git a/Objects/genobject.c b/Objects/genobject.c
@@ -1598,6 +1598,12 @@ static PyMethodDef async_gen_asend_methods[] = {
 };
 
 
+static PyMemberDef async_gen_asend_memberlist[] = {
+    {"ags_gen", T_OBJECT, offsetof(PyAsyncGenASend, ags_gen), READONLY},
+    {NULL}      /* Sentinel */
+};
+
+
 static PyAsyncMethods async_gen_asend_as_async = {
     PyObject_SelfIter,                          /* am_await */
     0,                                          /* am_aiter */
@@ -1635,7 +1641,7 @@ PyTypeObject _PyAsyncGenASend_Type = {
     PyObject_SelfIter,                          /* tp_iter */
     (iternextfunc)async_gen_asend_iternext,     /* tp_iternext */
     async_gen_asend_methods,                    /* tp_methods */
-    0,                                          /* tp_members */
+    async_gen_asend_memberlist,                 /* tp_members */
     0,                                          /* tp_getset */
     0,                                          /* tp_base */
     0,                                          /* tp_dict */
@@ -1951,6 +1957,12 @@ static PyMethodDef async_gen_athrow_methods[] = {
 };
 
 
+static PyMemberDef async_gen_athrow_memberlist[] = {
+    {"agt_gen", T_OBJECT, offsetof(PyAsyncGenAThrow, agt_gen), READONLY},
+    {NULL}      /* Sentinel */
+};
+
+
 static PyAsyncMethods async_gen_athrow_as_async = {
     PyObject_SelfIter,                          /* am_await */
     0,                                          /* am_aiter */
@@ -1988,7 +2000,7 @@ PyTypeObject _PyAsyncGenAThrow_Type = {
     PyObject_SelfIter,                          /* tp_iter */
     (iternextfunc)async_gen_athrow_iternext,    /* tp_iternext */
     async_gen_athrow_methods,                   /* tp_methods */
-    0,                                          /* tp_members */
+    async_gen_athrow_memberlist,                /* tp_members */
     0,                                          /* tp_getset */
     0,                                          /* tp_base */
     0,                                          /* tp_dict */

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	+Allow accessing the ``ags_gen`` and ``agt_gen`` attributes of asynchronous
	`2`	`+generators.`