make adding a single module simpler for interpreter users

youknowone · youknowone · commit 9220b2ed4e61 · 2023-03-31T00:31:45.000+09:00
diff --git a/src/interpreter.rs b/src/interpreter.rs
@@ -29,6 +29,16 @@ impl InterpreterConfig {
         self.init_hooks.push(hook);
         self
     }
+    pub fn add_native_module(
+        self,
+        name: impl ToOwned<Owned = String>,
+        make_module: fn(&VirtualMachine) -> rustpython_vm::PyObjectRef,
+    ) -> Self {
+        let name = name.to_owned();
+        self.init_hook(Box::new(move |vm| {
+            vm.add_native_module(name, Box::new(make_module))
+        }))
+    }
     #[cfg(feature = "stdlib")]
     pub fn init_stdlib(self) -> Self {
         self.init_hook(Box::new(init_stdlib))
