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

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

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

2. 配置Cargo.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"] }
   

3. 编辑src/lib.rs文件

   #![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(())	// 返回成功结果，表示模块初始化成功
   }
   

4. 命令行测试Python安装成功，python --version，pyo3 0.20.0支持的python版本为3.7-3.11

5. 命令行构建cargo build --release

6. 在target/release找到rust_python_ext.dll，复制到Python项目中，将后缀改为.pyd

7. 保证rust_python_ext.pyd和调用的py文件在同一目录下，在其中调用

   import rust_python_ext
   print(rust_python_ext.add(1,2))