Feb 23, 2023 · Jan 4, 2023 · Jan 4, 2023 · Jan 4, 2023 · Jan 4, 2023 · Feb 13, 2023
diff --git a/Include/cpython/genobject.h b/Include/cpython/genobject.h
 PyAPI_FUNC(int) _PyGen_SetStopIterationValue(PyObject *);
 PyAPI_FUNC(int) _PyGen_FetchStopIterationValue(PyObject **);
 PyAPI_FUNC(void) _PyGen_Finalize(PyObject *self);
 PyAPI_FUNC(PyCodeObject *) PyGen_GetCode(PyGenObject *gen);


 /* --- PyCoroObject ------------------------------------------------------- */
diff --git a/Lib/test/test_capi/test_misc.py b/Lib/test/test_capi/test_misc.py
        self.pendingcalls_submit(l, n)
        self.pendingcalls_wait(l, n)

    def test_gen_get_code(self):
       def genf(): yield
       gen = genf()
       self.assertEqual(_testcapi.gen_get_code(gen), gen.gi_code)


 class SubinterpreterTest(unittest.TestCase):

diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c
    return PyUnicode_FromString(PyEval_GetFuncDesc(func));
 }

 static PyObject *
 gen_get_code(PyObject *self, PyObject *gen)
 {
    if (!PyGen_Check(gen)) {
        PyErr_SetString(PyExc_TypeError, "argument must be a generator object");
        return NULL;
    }
    return (PyObject *)PyGen_GetCode((PyGenObject *)gen);
 }

 static PyObject *
 get_feature_macros(PyObject *self, PyObject *Py_UNUSED(args))
 {
    {"frame_getvarstring", test_frame_getvarstring, METH_VARARGS, NULL},
    {"eval_get_func_name", eval_get_func_name, METH_O, NULL},
    {"eval_get_func_desc", eval_get_func_desc, METH_O, NULL},
    {"gen_get_code", gen_get_code, METH_O, NULL},
    {"get_feature_macros", get_feature_macros, METH_NOARGS, NULL},
    {"test_code_api", test_code_api, METH_NOARGS, NULL},
    {"settrace_to_record", settrace_to_record, METH_O, NULL},
diff --git a/Objects/genobject.c b/Objects/genobject.c
    return frame->f_code;
 }

 PyCodeObject *
 PyGen_GetCode(PyGenObject *gen) {
    assert(PyGen_Check(gen));
    PyCodeObject *res = _PyGen_GetCode(gen);
    Py_INCREF(res);
    return res;
 }

 static inline int
 exc_state_traverse(_PyErr_StackItem *exc_state, visitproc visit, void *arg)
 {
Original file line number	Diff line number	Diff line change
Expand Up		@@ -44,6 +44,7 @@ PyAPI_FUNC(PyObject ) PyGen_NewWithQualName(PyFrameObject ,
		PyAPI_FUNC(int) _PyGen_SetStopIterationValue(PyObject *);
		PyAPI_FUNC(int) _PyGen_FetchStopIterationValue(PyObject **);
		PyAPI_FUNC(void) _PyGen_Finalize(PyObject *self);
		PyAPI_FUNC(PyCodeObject ) PyGen_GetCode(PyGenObject gen);
ericsnowcurrently marked this conversation as resolved. Show resolvedHide resolved


		/* --- PyCoroObject ------------------------------------------------------- */
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -1214,6 +1214,11 @@ def test_pendingcalls_non_threaded(self):
		self.pendingcalls_submit(l, n)
		self.pendingcalls_wait(l, n)

		def test_gen_get_code(self):
		def genf(): yield
		gen = genf()
		self.assertEqual(_testcapi.gen_get_code(gen), gen.gi_code)


		class SubinterpreterTest(unittest.TestCase):

Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2943,6 +2943,16 @@ eval_get_func_desc(PyObject self, PyObject func)
		return PyUnicode_FromString(PyEval_GetFuncDesc(func));
		}

		static PyObject *
		gen_get_code(PyObject self, PyObject gen)
		{
		if (!PyGen_Check(gen)) {
		PyErr_SetString(PyExc_TypeError, "argument must be a generator object");
		return NULL;
		}
		return (PyObject )PyGen_GetCode((PyGenObject )gen);
		}

		static PyObject *
		get_feature_macros(PyObject self, PyObject Py_UNUSED(args))
		{
Expand DownExpand Up		@@ -3346,6 +3356,7 @@ static PyMethodDef TestMethods[] = {
		{"frame_getvarstring", test_frame_getvarstring, METH_VARARGS, NULL},
		{"eval_get_func_name", eval_get_func_name, METH_O, NULL},
		{"eval_get_func_desc", eval_get_func_desc, METH_O, NULL},
		{"gen_get_code", gen_get_code, METH_O, NULL},
		{"get_feature_macros", get_feature_macros, METH_NOARGS, NULL},
		{"test_code_api", test_code_api, METH_NOARGS, NULL},
		{"settrace_to_record", settrace_to_record, METH_O, NULL},
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -31,6 +31,14 @@ _PyGen_GetCode(PyGenObject *gen) {
		return frame->f_code;
		}

		PyCodeObject *
		PyGen_GetCode(PyGenObject *gen) {
		assert(PyGen_Check(gen));
		PyCodeObject *res = _PyGen_GetCode(gen);
		Py_INCREF(res);
		return res;
		}

		static inline int
		exc_state_traverse(_PyErr_StackItem exc_state, visitproc visit, void arg)
		{
Expand Down