pyframe fix 9-test

Author: Yashp002

Doc/c-api/frame.rst

@@ -274,7 +274,7 @@ Unless using :pep:`523`, you will not need this.
 
    .. code-block:: c
 
-      int kind = PyUnstable_Frame_GetExecutableKind(frame)
+      int kind = PyUnstable_Frame_GetExecutableKind(frame);
 
       if (kind == PyUnstable_EXECUTABLE_KIND_SKIP) {
           continue;

Include/cpython/frameobject.h

@@ -24,6 +24,7 @@ PyAPI_FUNC(void) PyFrame_LocalsToFast(PyFrameObject *, int);
  * obsolete or outdated. */
 
 PyAPI_FUNC(int) _PyFrame_IsEntryFrame(PyFrameObject *frame);
+PyAPI_FUNC(int) PyUnstable_Frame_GetExecutableKind(PyFrameObject *frame);
 
 PyAPI_FUNC(int) PyFrame_FastToLocalsWithError(PyFrameObject *f);
 PyAPI_FUNC(void) PyFrame_FastToLocals(PyFrameObject *);

Modules/_testcapi/frame.c

@@ -114,7 +114,26 @@ frame_getvarstring(PyObject *self, PyObject *args)
 }
 
 
+static PyObject *
+test_my_doc_example(PyObject *self, PyObject *arg)
+{
+    PyFrameObject *frame = (PyFrameObject *)PyEval_GetFrame();
+    if (frame == NULL) {
+        Py_RETURN_NONE;
+    }
+
+    int kind = PyUnstable_Frame_GetExecutableKind(frame);
+
+    if (kind == PyUnstable_EXECUTABLE_KIND_SKIP) {
+         return PyLong_FromLong(kind);
+    }
+
+    return PyLong_FromLong(kind);
+}
+
+
 static PyMethodDef test_methods[] = {
+    {"test_my_doc_example", test_my_doc_example, METH_NOARGS, NULL},
     {"frame_getlocals", frame_getlocals, METH_O, NULL},
     {"frame_getglobals", frame_getglobals, METH_O, NULL},
     {"frame_getgenerator", frame_getgenerator, METH_O, NULL},
@@ -131,4 +150,3 @@ _PyTestCapi_Init_Frame(PyObject *m)
 {
     return PyModule_AddFunctions(m, test_methods);
 }
-

Python/frame.c

@@ -157,3 +157,23 @@ const PyTypeObject *const PyUnstable_ExecutableKinds[PyUnstable_EXECUTABLE_KINDS
     [PyUnstable_EXECUTABLE_KIND_METHOD_DESCRIPTOR] = &PyMethodDescr_Type,
     [PyUnstable_EXECUTABLE_KINDS] = NULL,
 };
+
+PyAPI_FUNC(int)
+PyUnstable_Frame_GetExecutableKind(PyFrameObject *frame)
+{
+    _PyInterpreterFrame *f = frame->f_frame;
+
+    PyObject *exec = PyStackRef_AsPyObjectBorrow(f->f_executable);
+
+    if (PyCode_Check(exec)) {
+        return PyUnstable_EXECUTABLE_KIND_PY_FUNCTION;
+    }
+    if (PyMethod_Check(exec)) {
+        return PyUnstable_EXECUTABLE_KIND_BUILTIN_FUNCTION;
+    }
+    if (Py_IS_TYPE(exec, &PyMethodDescr_Type)) {
+        return PyUnstable_EXECUTABLE_KIND_METHOD_DESCRIPTOR;
+    }
+
+    return PyUnstable_EXECUTABLE_KIND_SKIP;
+}