如何把python代码打包成.so文件

一、使用 Cython 打包 Python 代码

1、安装Cython

pip install cython

2、编写python代码

假设你有一个 example.py文件：

# example.pyx
def say_hello(name):
    print(f"Hello, {name}!")

3、创建 setup.py 文件:

# setup.py
from setuptools import setup
from Cython.Build import cythonize

setup(
    ext_modules=cythonize("example.pyx")
)

4、编译 .py 文件:

运行以下命令以生成 .so 文件：

python setup.py build_ext --inplace

这会生成一个名为 example.cpython-<version>-<platform>.so 的文件。

二、使用 C++ 和 Python API

如果你需要更复杂的打包方式，涉及到 C++ 和 Python API，步骤会更复杂，通常包括创建 C++ 代码和使用 pybind11 等工具来包装 Python 接口。

1、安装pybind11

pip install pybind11

2、编写C++代码

// example.cpp
#include <pybind11/pybind11.h>

void say_hello(const std::string &name) {
    std::cout << "Hello, " << name << "!" << std::endl;
}

PYBIND11_MODULE(example, m) {
    m.def("say_hello", &say_hello);
}

3、创建 CMakeLists.txt 文件:

cmake_minimum_required(VERSION 3.0)
project(example)

find_package(pybind11 REQUIRED)

pybind11_add_module(example example.cpp)

4、编译 C++ 代码:

mkdir build
cd build
cmake ..
make

这将生成一个 example.so 文件。

这样，你可以在 Python 中通过 import example 来使用这个 .so 文件。