Add pyframe functions to c-api (#8215)

bschoenmaeckers · web-flow · commit 28829a71d7d0 · 2026-07-07T00:03:43.000+09:00
diff --git a/.cspell.dict/cpython.txt b/.cspell.dict/cpython.txt
@@ -191,6 +191,7 @@ pycore
 pyinner
 pydecimal
 pyerrors
+pyframe
 Pyfunc
 pylifecycle
 pymain
diff --git a/crates/capi/src/lib.rs b/crates/capi/src/lib.rs
@@ -28,6 +28,7 @@ pub mod object;
 pub mod osmodule;
 pub mod pycapsule;
 pub mod pyerrors;
+pub mod pyframe;
 pub mod pylifecycle;
 pub mod pymem;
 pub mod pystate;
diff --git a/crates/capi/src/pyframe.rs b/crates/capi/src/pyframe.rs
@@ -0,0 +1,19 @@
+use crate::PyObject;
+use crate::pystate::with_vm;
+use rustpython_vm::frame::Frame;
+
+#[unsafe(no_mangle)]
+pub unsafe extern "C" fn PyFrame_GetCode(frame: *mut PyObject) -> *mut PyObject {
+    with_vm(|vm| {
+        let frame = unsafe { &*frame }.try_downcast_ref::<Frame>(vm)?;
+        Ok(frame.f_code())
+    })
+}
+
+#[unsafe(no_mangle)]
+pub unsafe extern "C" fn PyFrame_GetLineNumber(frame: *mut PyObject) -> core::ffi::c_int {
+    with_vm(|vm| {
+        let frame = unsafe { &*frame }.try_downcast_ref::<Frame>(vm)?;
+        Ok(frame.f_lineno() as core::ffi::c_int)
+    })
+}
