当前位置：首页 > news >正文

Milksnake与Cargo完美配合：Rust开发者的Python扩展指南

news 2026/6/18 8:31:08

Milksnake与Cargo完美配合：Rust开发者的Python扩展指南

【免费下载链接】milksnakeA setuptools/wheel/cffi extension to embed a binary data in wheels项目地址: https://gitcode.com/gh_mirrors/mi/milksnake

Milksnake是一款强大的setuptools扩展工具，专为Rust开发者打造，可轻松将Rust编写的动态链接库嵌入Python Wheel包中，实现跨平台分发。本文将详细介绍如何通过Milksnake与Cargo的无缝协作，快速构建高性能的Python扩展模块。

🚀 为什么选择Milksnake？

传统Python扩展开发常面临跨平台兼容性和构建流程复杂的问题。Milksnake通过创新的工作流解决了这些痛点：

跨平台支持：一次构建即可支持Linux、Mac和Windows系统
Cargo集成：直接调用Rust的Cargo构建系统，保留Rust生态优势
CFFI桥接：通过CFFI实现Python与Rust的高效交互，无需手动编写绑定代码
通用Wheel包：生成py2.py3-none-PLATFORM.whl格式的通用Wheel，减少维护成本

与其他工具相比，Milksnake专注于将原生库与Python无缝集成，特别适合需要高性能计算的场景。

📋 快速开始：环境准备

在开始前，请确保系统已安装以下工具：

Python 3.6+ 及对应的setuptools
Rust 1.40+ 及Cargo包管理器
pip 19.0+（用于Wheel安装）

通过以下命令克隆项目仓库：

git clone https://gitcode.com/gh_mirrors/mi/milksnake cd milksnake

🔧 核心配置：setup.py详解

Milksnake的核心在于setup.py中的配置，它定义了Rust代码的构建流程和Python模块的桥接方式。典型配置如下：

from setuptools import setup def build_native(spec): # 配置Cargo构建命令 build = spec.add_external_build( cmd=['cargo', 'build', '--release'], path='./rust' ) # 配置CFFI模块 spec.add_cffi_module( module_path='example._native', dylib=lambda: build.find_dylib('example', in_path='target/release'), header_filename=lambda: build.find_header('example.h', in_path='target'), rtld_flags=['NOW', 'NODELETE'] ) setup( name='example', version='0.0.1', packages=['example'], zip_safe=False, platforms='any', setup_requires=['milksnake'], install_requires=['milksnake'], milksnake_tasks=[build_native] # 注册Milksnake构建任务 )

关键配置说明：

add_external_build：定义外部构建命令（这里是Cargo）
add_cffi_module：指定CFFI模块的路径和动态库信息
milksnake_tasks：注册构建任务，这是Milksnake的核心入口点

🦀 Rust库结构与配置

Cargo.toml配置

Rust库需要配置为动态链接库（cdylib）类型，典型的Cargo.toml如下：

[package] name = "example" version = "0.1.0" build = "build.rs" [lib] name = "example" crate-type = ["cdylib"] # 生成C兼容的动态链接库 [build-dependencies] cbindgen = "0.4" # 用于生成C头文件

Rust源代码实现

在rust/src/lib.rs中实现核心功能：

#[no_mangle] pub unsafe extern "C" fn a_function_from_rust() -> i32 { 42 // 简单返回整数42作为示例 }

头文件生成

通过build.rs使用cbindgen自动生成C头文件：

extern crate cbindgen; use std::env; fn main() { let crate_dir = env::var("CARGO_MANIFEST_DIR").unwrap(); let mut config = cbindgen::Config::default(); config.language = cbindgen::Language::C; cbindgen::generate_with_config(&crate_dir, config) .unwrap() .write_to_file("target/example.h"); }

🐍 Python模块桥接

创建Python包结构，在example/__init__.py中通过CFFI调用Rust函数：

from example._native import ffi, lib def test(): return lib.a_function_from_rust() # 调用Rust实现的函数

🏗️ 构建与安装流程

构建Wheel包

通过以下命令构建Python Wheel包：

python setup.py bdist_wheel

Milksnake会自动调用Cargo构建Rust代码，并将生成的动态库打包到Wheel中。

安装与测试

安装生成的Wheel包：

pip install dist/example-0.0.1-py2.py3-none-*.whl

在Python中测试：

import example print(example.test()) # 输出: 42

📁 项目结构最佳实践

推荐的项目结构如下：

project/ ├── example/ # Python包目录 │ ├── __init__.py │ └── _native.py # CFFI生成的模块 ├── rust/ # Rust库目录 │ ├── src/ │ │ └── lib.rs │ ├── Cargo.toml │ └── build.rs └── setup.py # Milksnake配置

这种结构清晰分离了Python和Rust代码，便于维护和扩展。