最简单的解决方案与pybind11本身无关。当作者想要在同一个包中结合纯 Python 和 C/Cython/其他本机扩展时,他们通常会做什么,如下所示。
您创建两个模块。
mymodule
是一个公共接口,一个纯 Python 模块
_mymodule
是一个私有实现,一个编译模块
然后在mymodule
你导入必要的符号_mymoudle
(如果需要,回退到纯 Python 版本)。
这是来自yarl包的示例:
引用.py
try:
from ._quoting import _quote, _unquote
quote = _quote
unquote = _unquote
except ImportError: # pragma: no cover
quote = _py_quote
unquote = _py_unquote
_quoting.pyx
更新
下面是脚本。为了可重复性,我正在针对原始cmake_example进行操作。
git clone --recursive https://github.com/pybind/cmake_example.git
# at the time of writing https://github.com/pybind/cmake_example/commit/8818f493
cd cmake_example
现在创建纯 Python 模块(内部cmake_example/cmake_example
)。
cmake_example/__init__.py
"""Root module of your package"""
cmake_example/math.py
def mul(a, b):
"""Pure Python-only function"""
return a * b
def add(a, b):
"""Fallback function"""
return a + b
try:
from ._math import add
except ImportError:
pass
现在让我们修改现有文件以将cmake_example
模块转换为cmake_example._math
.
src/main.cpp
(subtract
为简洁起见已删除)
#include <pybind11/pybind11.h>
int add(int i, int j) {
return i + j;
}
namespace py = pybind11;
PYBIND11_MODULE(_math, m) {
m.doc() = R"pbdoc(
Pybind11 example plugin
-----------------------
.. currentmodule:: _math
.. autosummary::
:toctree: _generate
add
)pbdoc";
m.def("add", &add, R"pbdoc(
Add two numbers
Some other explanation about the add function.
)pbdoc");
#ifdef VERSION_INFO
m.attr("__version__") = VERSION_INFO;
#else
m.attr("__version__") = "dev";
#endif
}
CMakeLists.txt
cmake_minimum_required(VERSION 2.8.12)
project(cmake_example)
add_subdirectory(pybind11)
pybind11_add_module(_math src/main.cpp)
setup.py
# the above stays intact
from subprocess import CalledProcessError
kwargs = dict(
name='cmake_example',
version='0.0.1',
author='Dean Moldovan',
author_email='dean0x7d@gmail.com',
description='A test project using pybind11 and CMake',
long_description='',
ext_modules=[CMakeExtension('cmake_example._math')],
cmdclass=dict(build_ext=CMakeBuild),
zip_safe=False,
packages=['cmake_example']
)
# likely there are more exceptions, take a look at yarl example
try:
setup(**kwargs)
except CalledProcessError:
print('Failed to build extension!')
del kwargs['ext_modules']
setup(**kwargs)
现在我们可以构建它了。
python setup.py bdist_wheel
在我的情况下,它会产生dist/cmake_example-0.0.1-cp27-cp27mu-linux_x86_64.whl
(如果 C++ 编译失败它是cmake_example-0.0.1-py2-none-any.whl
)。这是它的内容(unzip -l ...
):
Archive: cmake_example-0.0.1-cp27-cp27mu-linux_x86_64.whl
Length Date Time Name
--------- ---------- ----- ----
0 2017-12-05 21:42 cmake_example/__init__.py
81088 2017-12-05 21:43 cmake_example/_math.so
223 2017-12-05 21:46 cmake_example/math.py
10 2017-12-05 21:48 cmake_example-0.0.1.dist-info/DESCRIPTION.rst
343 2017-12-05 21:48 cmake_example-0.0.1.dist-info/metadata.json
14 2017-12-05 21:48 cmake_example-0.0.1.dist-info/top_level.txt
105 2017-12-05 21:48 cmake_example-0.0.1.dist-info/WHEEL
226 2017-12-05 21:48 cmake_example-0.0.1.dist-info/METADATA
766 2017-12-05 21:48 cmake_example-0.0.1.dist-info/RECORD
--------- -------
82775 9 files