Question

我有以下目录布局

awesome_package
\- module1.py
\- build
   \- module2.so

我目前将module1导入为

import awesome_package.module1

和module2为

import sys
sys.path.append('path/to/awesome_package/build')
import module2

但我希望能够使用以前的语法。

module2由pybind11以如下方式创建：

PYBIND11_MODULE(module2, module2) {
    module2.doc() = "C++ module wrapped for Python";
    module2.def("some_cpp_function", some_cpp_function) 
}

Answer 1

正如我在评论中所说，二进制Python模块通常使用distutils / setuptools构建。为此，您需要使用所有必要选项编写setup.py脚本。以下是一个非常简单的示例，仅显示基本内容：

from setuptools import setup, Extension

setup(
    name = 'awesome',
    version = '0.0.1',
    packages = ['awesome_package']                     
    ext_modules = [Extension(
       'awesome_package.module2',
       ['src/module2.cpp']
    )]
)

当然，在setup.py中，您需要指定所有构建选项，如头文件，编译器标志等。

如果您坚持使用CMake，则需要添加自定义复制命令以复制包中的已编译模块。像这样：

add_custom_command(TARGET module2 POST_BUILD
       COMMAND ${CMAKE_COMMAND} -E copy $<TARGET_FILE:module2>
       "${CMAKE_SOURCE_DIR}/awesome_package"
    )

使C ++模块成为Python包的一部分

1 个答案: