Rust 使用RustPython-JavaScript中文网-JavaScript教程资源分享门户

在本章中，我们将深入探讨如何使用 RustPython 在 Rust 项目中嵌入 Python 解释器。RustPython 是一个用 Rust 编写的 Python 解释器，它允许你在 Rust 应用程序中直接运行 Python 代码，从而为 Rust 程序员提供了极大的灵活性。

安装 RustPython

首先，你需要安装 Rust 和 Cargo。如果你已经安装了 Rust 和 Cargo，可以跳过这一步。打开终端并输入以下命令来安装 Rust：

curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh

安装完成后，你需要更新你的 PATH 变量，以便可以访问 cargo 工具。根据安装过程中的提示完成此步骤。

接下来，你需要将 RustPython 添加到你的 Cargo 项目中。打开你的 Cargo.toml 文件，并添加以下依赖项：

[dependencies]
rustpython-vm = "0.4"

保存文件后，你可以通过运行 cargo build 来安装这个依赖项。

创建一个简单的 Rust 项目

假设你已经有了一个 Rust 项目，我们可以通过以下步骤来设置项目：

打开你的项目目录。
编辑 Cargo.toml 文件，添加 RustPython 依赖项。
在 src/main.rs 中添加如下代码：

use rustpython_vm::VirtualMachine;

fn main() {
    let vm = VirtualMachine::default();
    let code = r#"print("Hello, RustPython!")"#;
    vm.run_code_obj(vm.compile(code, "<string>").unwrap()).unwrap();
}

这段代码创建了一个虚拟机实例，并执行了一段 Python 代码，输出 "Hello, RustPython!"。

运行和调试你的项目

保存所有更改后，运行以下命令来编译和运行你的项目：

cargo run

你应该会在终端中看到输出 "Hello, RustPython!"。

在 Rust 中调用 Python 函数

接下来，我们将演示如何在 Rust 中调用 Python 函数。假设你有一个简单的 Python 脚本，名为 script.py，内容如下：

def greet(name):
    return f"Hello, {name}!"

print(greet("World"))

你可以在 Rust 中调用这个函数，如下所示：

首先，确保你的 script.py 文件位于项目的根目录中。
修改 src/main.rs 文件，使其包含以下内容：

// www.javascriptcn.com code example
use rustpython_vm::{VirtualMachine, PyResult};

fn main() -> PyResult<()> {
    let vm = VirtualMachine::default();

    // 加载 Python 脚本
    let script_code = std::fs::read_to_string("script.py")?;
    vm.run_code_obj(vm.compile(&script_code, "<script>").unwrap())?;

    // 调用 Python 函数
    let locals = vm.new_scope_with_builtins();
    let py_greet = vm.get_attribute(locals.get("greet").unwrap(), "greet").unwrap();
    let result = vm.invoke(py_greet, (vm.new_str("Rust".to_string()),))?;
    println!("{}", vm.to_pyobj(result));

    Ok(())
}

运行项目：

cargo run

你应该会看到输出 "Hello, Rust!"。

处理异常

在处理 Python 代码时，可能会遇到各种异常情况。你可以通过 Rust 的错误处理机制来捕获这些异常。例如，你可以修改上面的代码来处理可能出现的异常：

// www.javascriptcn.com code example
use rustpython_vm::{VirtualMachine, PyResult};

fn main() -> PyResult<()> {
    let vm = VirtualMachine::default();

    // 加载 Python 脚本
    let script_code = std::fs::read_to_string("script.py")?;
    match vm.run_code_obj(vm.compile(&script_code, "<script>").unwrap()) {
        Ok(_) => (),
        Err(e) => {
            eprintln!("Error running script: {:?}", e);
            return Err(e);
        }
    }

    // 调用 Python 函数
    let locals = vm.new_scope_with_builtins();
    let py_greet = vm.get_attribute(locals.get("greet").unwrap(), "greet").unwrap();
    match vm.invoke(py_greet, (vm.new_str("Rust".to_string()),)) {
        Ok(result) => println!("{}", vm.to_pyobj(result)),
        Err(e) => {
            eprintln!("Error invoking function: {:?}", e);
        }
    }

    Ok(())
}

这样，如果在运行脚本或调用函数时发生异常，程序将打印出相应的错误信息。

结合 Rust 和 Python 的数据结构

在某些情况下，你可能需要在 Rust 和 Python 之间传递复杂的数据结构。RustPython 提供了一些方法来实现这一点。下面是一个示例，展示了如何从 Rust 向 Python 传递一个字典：

// www.javascriptcn.com code example
use rustpython_vm::{VirtualMachine, PyResult, PyObject};
use rustpython_vm::types::PyDict;

fn main() -> PyResult<()> {
    let vm = VirtualMachine::default();

    // 创建一个 Python 字典
    let py_dict = PyDict::new(&vm);
    py_dict.set_item(&vm, "key", "value").unwrap();

    // 将字典传递给 Python 代码
    let locals = vm.new_scope_with_builtins();
    locals.set_item("my_dict", py_dict.into_object()).unwrap();

    let code = r#"
def print_dict(d):
    for key, value in d.items():
        print(f"{key}: {value}")

print_dict(my_dict)
"#;

    vm.run_code_obj(vm.compile(code, "<string>").unwrap())?;

    Ok(())
}

在这个例子中，我们创建了一个 Rust 的 PyDict 对象，并将其传递给 Python 代码，然后在 Python 中遍历并打印字典的内容。

总结

通过本章的学习，你应该已经掌握了如何在 Rust 项目中使用 RustPython 来嵌入 Python 解释器，并能够处理基本的 Python 代码和数据结构。RustPython 为 Rust 开发者提供了一个强大的工具，使得在 Rust 应用程序中集成 Python 成为可能。希望这些知识对你今后的工作有所帮助！