Rust学习日记：编写一个Python扩展

m0_521118232025-04-06 16:26

参考https://segmentfault.com/a/1190000044555330

命令行创建一个新的Rust项目cargo new --lib rust_python_ext

配置Cargo.toml

toml 复制代码

[package]
name = "rust_python_ext"
version = "0.1.0"
edition = "2024"

[lib]
name = "rust_python_ext"
crate-type = ["cdylib"]

[dependencies]
pyo3 = { version = "0.20.0", features = ["extension-module", "macros"] }

编辑src/lib.rs文件

rust 复制代码

#![allow(unsafe_op_in_unsafe_fn)]

use pyo3::prelude::*;

#[pyfunction]	// 定义一个Python可调用的函数
// i32: 32位整数
fn add(a: i32, b: i32) -> i32 {
    a + b
}
// 重点：函数名必须和 .pyd 文件名一致（这里是 `rust_python_ext`）
#[pymodule]		// 定义一个Python模块
// 模块初始化函数，接收Python解释器实例和模块对象，且名字必须与项目名一致
// _py：Python解释器实例(下划线表示未使用)
// m:表示初始化的模块对象
// &表示不可变引用
// PyResult<()> 表示一个可能返回Python异常或成功的操作结果
fn rust_python_ext(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_function(wrap_pyfunction!(add, m)?)?;
    // ?用于错误处理
    Ok(())	// 返回成功结果，表示模块初始化成功
}

命令行测试Python安装成功，python --version，pyo3 0.20.0支持的python版本为3.7-3.11
命令行构建cargo build --release
在target/release找到rust_python_ext.dll，复制到Python项目中，将后缀改为.pyd
保证rust_python_ext.pyd和调用的py文件在同一目录下，在其中调用
python 复制代码
```
import rust_python_ext
print(rust_python_ext.add(1,2))
```