RustPython/examples/freeze/main.rs

use std::collections::HashMap;

use rustpython_vm as vm;

fn main() -> vm::pyobject::PyResult<()> {
    vm::Interpreter::default().enter(run)
}

fn run(vm: &vm::VirtualMachine) -> vm::pyobject::PyResult<()> {
    let scope = vm.new_scope_with_builtins();

    // the file parameter is relevant to the directory where the crate's Cargo.toml is located, see $CARGO_MANIFEST_DIR:
    // https://doc.rust-lang.org/cargo/reference/environment-variables.html#environment-variables-cargo-sets-for-crates
    let modules: HashMap<String, vm::bytecode::FrozenModule> =
        vm::py_freeze!(file = "examples/freeze/freeze.py");

    let res = vm.run_code_obj(
        vm.new_code_object(modules.get("frozen").unwrap().code.clone()),
        scope,
    );

    if let Err(err) = res {
        vm::exceptions::print_exception(&vm, err);
    }

    Ok(())
}