如何指定 pyo3 在 rust 项目中使用的 virtualenv？答案

【问题标题】：How to specify to pyo3 the virtualenv to use in a rust project?如何指定 pyo3 在 rust 项目中使用的 virtualenv？
【发布时间】：2022-12-04 17:11:22
【问题描述】：

我需要在我的 rust 项目中移植一些 python 代码（从 rust 调用 python）。

我正在用 Rust 编写一个应用程序，其中一小部分需要导入一个用 Python 编写的模块。

这是我的项目结构。

|...
|extern/python/
|-main.py
|-__init__.py
|src/
|Cargo.toml
|...

我对从 rust 执行 python 代码没有问题，pyo3 的文档涵盖了这一点。

但是我需要向 pyo3 指定 virtualenv 用来链接到我的 Rust crate，几乎所有 pyo3 的文档都集中在如何从 python 使用 rust，而关于如何从 rust 使用 python 的信息很少。

在我的 python 代码中，y 使用 pyenv 来隔离 virtualenv，例如运行 python 代码

cd /python/code
pyenv shell my_py_env
python3 main.py

所以问题是，我如何告诉 pyo3 在链接 python 模块时使用 "my_py_env"？

【问题讨论】：

标签： python rust ffi pyo3

【解决方案1】：

要指定在 Rust 项目中导入 Python 模块时 pyo3 应该使用哪个虚拟环境，可以使用 PYTHONHOME 和 PYTHONPATH 环境变量。

下面是一个示例，说明在导入 Python 模块之前如何在 Rust 代码中设置这些环境变量：

use std::env;
use pyo3::prelude::*;

fn main() -> PyResult<()> {
    // Set the PYTHONHOME and PYTHONPATH environment variables
    env::set_var("PYTHONHOME", "/path/to/pyenv/versions/my_py_env");
    env::set_var("PYTHONPATH", "/path/to/extern/python");

    // Import the Python module
    let gil = Python::acquire_gil();
    let py = gil.python();
    let py_module = PyModule::from_code(py, include_bytes!("../extern/python/main.py"), "main.py", "main")?;

    // Use the imported module...

    Ok(())
}

本例中PYTHONHOME变量设置为my_py_env虚拟环境的路径，PYTHONPATH变量设置为Python模块所在extern/python目录的路径。

一旦你设置了这些环境变量，pyo3 将在你的 Rust 代码中导入 Python 模块时使用 my_py_env 虚拟环境。

【讨论】：