RustPython/examples/package_embed.rs at main · RustPython/RustPython · GitHub

Repository navigation

30 lines (28 loc) · 1.09 KB

use rustpython_vm as vm;

use std::process::ExitCode;

use vm::{Interpreter, builtins::PyStrRef};

fn py_main(interp: &Interpreter) -> vm::PyResult<PyStrRef> {

interp.enter(|vm| {

// Add local library path

vm.insert_sys_path(vm.new_pyobj("examples"))

.expect("add examples to sys.path failed");

let module = vm.import("package_embed", 0)?;

let name_func = module.get_attr("context", vm)?;

let result = name_func.call((), vm)?;

let result: PyStrRef = result.get_attr("name", vm)?.try_into_value(vm)?;

vm::PyResult::Ok(result)

})

}

fn main() -> ExitCode {

// Add standard library path

let mut settings = vm::Settings::default();

settings.path_list.push("Lib".to_owned());

let builder = vm::Interpreter::builder(settings);

let defs = rustpython_stdlib::stdlib_module_defs(&builder.ctx);

let interp = builder.add_native_modules(&defs).build();

let result = py_main(&interp);

let result = result.map(|result| {

println!("name: {result}");

});

vm::host_env::os::exit_code(interp.run(|_vm| result))

}