Add import support

bschoenmaeckers · bschoenmaeckers · commit 2ef64d2578e2 · 2026-04-08T15:48:31.000+02:00
diff --git a/crates/capi/src/import.rs b/crates/capi/src/import.rs
@@ -1,9 +1,17 @@
-use core::ptr;
-
 use crate::PyObject;
+use crate::pystate::with_vm;
+use rustpython_vm::builtins::PyStr;
 
 #[unsafe(no_mangle)]
-pub extern "C" fn PyImport_Import(_name: *mut PyObject) -> *mut PyObject {
-    crate::log_stub("PyImport_Import");
-    ptr::null_mut()
+pub extern "C" fn PyImport_Import(name: *mut PyObject) -> *mut PyObject {
+    with_vm(|vm| {
+        let name = unsafe { (&*name).downcast_unchecked_ref::<PyStr>() };
+        vm.import(name, 0).map_or_else(
+            |err| {
+                vm.push_exception(Some(err));
+                std::ptr::null_mut()
+            },
+            |module| module.into_raw().as_ptr(),
+        )
+    })
 }
diff --git a/example_projects/pyo3_embed/src/main.rs b/example_projects/pyo3_embed/src/main.rs
@@ -39,6 +39,8 @@ fn main() {
         PyTypeError::new_err("This is a type error").restore(py);
         assert!(PyErr::take(py).is_some());
 
+        py.import("sys")?;
+
         PyResult::Ok(())
     })
     .unwrap();
