Rust使用PyO3时如何让返回值不绑定Python解释器的生命周期

问题根源 #

你遇到的生命周期报错核心原因是你给泛型T绑定的生命周期'a是函数级别的，而Python::with_gil闭包内所有Python对象的生命周期都和闭包内的GIL令牌绑定，生命周期短于函数级别的'a，编译器判定FromPyObject<'a>的约束无法被满足。

解决方法 #

使用Rust的高阶trait约束（HRTB）声明T可以从任意生命周期的Python对象中提取，不需要绑定固定的'a生命周期。同时你原有代码还存在两个逻辑问题：

1. py.eval的第一个参数没有动态替换func_name，直接写{}不会生效
2. 你代码中没有定义kwargs变量，执行eval时会抛出Python NameError

修改后的可运行代码如下：

use pyo3::{prelude::*, types::PyDict};

fn run_python<T: for<'a> FromPyObject<'a> + Clone>(func_name: &str, kwargs: Option<&PyDict>) -> Result<T, ()> {
    Python::with_gil(|py| {
        let pyapi = match py.import("pyapi") {
            Ok(v) => v,
            Err(e) => {
                e.print_and_set_sys_last_vars(py);
                return Err(());
            }
        };

        // 构造执行语句，动态插入函数名
        let eval_stmt = format!("pyapi.{}(**kwargs)", func_name);
        // 构造locals字典，同时传入pyapi和kwargs
        let locals = PyDict::new(py);
        locals.set_item("pyapi", pyapi).map_err(|_| ())?;
        locals.set_item("kwargs", kwargs.unwrap_or_else(|| PyDict::new(py))).map_err(|_| ())?;

        let eval_result: PyResult<&PyAny> = py.eval(&eval_stmt, None, Some(locals));

        let wrapped_obj: &PyAny = match eval_result {
            Ok(v) => v,
            Err(e) => {
                e.print_and_set_sys_last_vars(py);
                return Err(());
            }
        };

        let unwrapped_result: PyResult<T> = wrapped_obj.extract();

        match unwrapped_result {
            Ok(v) => Ok(v),
            Err(_) => Err(()),
        }
    })
}

关键修改说明 #

• 移除了函数定义中不必要的生命周期参数'a，改用for<'a> FromPyObject<'a>约束T：这个约束表示只要存在任意生命周期'a，T就可以从该生命周期的Python对象中提取值，提取完成后得到的T是纯Rust类型，完全脱离Python GIL和Python对象的生命周期约束，可以安全带出with_gil闭包。
• 补全了动态生成执行语句的逻辑，用format!把传入的函数名拼入执行代码
• 补全了kwargs参数的定义和传入逻辑，避免Python运行时报错
• 移除了不必要的clone()调用：extract()返回的T本身就是已经从Python对象中拷贝/转换出来的Rust值，不需要额外克隆。

内容的提问来源于stack exchange，提问作者Eli Stevens