11.2 通过PyPI发布使用CMake/pybind11构建的C++/Python项目

NOTE:此示例代码可以在 https://github.com/dev-cafe/cmake-cookbook/tree/v1.0/chapter-11/recipe-02 中找到。该示例在CMake 3.11版(或更高版本)中是有效的,并且已经在GNU/Linux、macOS和Windows上进行过测试。

本示例中,我们将以第9章第5节的代码的pybind11为例,为其添加相关的安装目标和pip打包信息,并将项目上传到PyPI。我们要实现一个可以使用pip安装,并运行CMake从而获取底层pybind11依赖项的项目。

准备工作

要通过PyPI分发包的话,需要一个https://pypi.org 帐户。当然,也可以先从本地路径进行安装练习。

TIPS:建议使用Pipenv (https://docs.pipenv.org )或虚拟环境(https://virtualenv.pypa )安装这个包和其他的Python包。

我们基于第9章第5节的项目,它包含一个主CMakeLists.txt文件和一个account/CMakeLists.txt文件,配置帐户示例目标时,使用如下的项目树:

  1. .
  2. ├── account
  3. ├── account.cpp
  4. ├── account.hpp
  5. ├── CMakeLists.txt
  6. s└── test.py
  7. └── CMakeLists.txt

示例中,account.cpp,account.hpptest.py没有任何变化。修改account/CMakeLists.txt,并为pip添加几个文件,以便能够构建安装包。为此,需要根目录中的另外三个文件:README.rstMANIFEST.insetup.py

README.rst中包含关于项目的s文档:

  1. Example project
  2. ===============
  3. Project description in here ...

MANIFEST.in列出了需要安装的Python模块:

  1. include README.rst CMakeLists.txt
  2. recursive-include account *.cpp *.hpp CMakeLists.txt

最后,setup.py包含构建指令和安装项目的说明:

  1. import distutils.command.build as _build
  2. import os
  3. import sys
  4. from distutils import spawn
  5. from distutils.sysconfig import get_python_lib
  6. from setuptools import setup
  7. def extend_build():
  8. class build(_build.build):
  9. def run(self):
  10. cwd = os.getcwd()
  11. if spawn.find_executable('cmake') is None:
  12. sys.stderr.write("CMake is required to build this package.\n")
  13. sys.exit(-1)
  14. _source_dir = os.path.split(__file__)[0]
  15. _build_dir = os.path.join(_source_dir, 'build_setup_py')
  16. _prefix = get_python_lib()
  17. try:
  18. cmake_configure_command = [
  19. 'cmake',
  20. '-H{0}'.format(_source_dir),
  21. '-B{0}'.format(_build_dir),
  22. '-DCMAKE_INSTALL_PREFIX={0}'.format(_prefix),
  23. ]
  24. _generator = os.getenv('CMAKE_GENERATOR')
  25. if _generator is not None:
  26. cmake_configure_command.append('-
  27. G{0}'.format(_generator))
  28. spawn.spawn(cmake_configure_command)
  29. spawn.spawn(
  30. ['cmake', '--build', _build_dir, '--target', 'install'])
  31. os.chdir(cwd)
  32. except spawn.DistutilsExecError:
  33. sys.stderr.write("Error while building with CMake\n")
  34. sys.exit(-1)
  35. _build.build.run(self)
  36. return build
  37. _here = os.path.abspath(os.path.dirname(__file__))
  38. if sys.version_info[0] < 3:
  39. with open(os.path.join(_here, 'README.rst')) as f:
  40. long_description = f.read()
  41. else:
  42. with open(os.path.join(_here, 'README.rst'), encoding='utf-8') as f:
  43. long_description = f.read()
  44. _this_package = 'account'
  45. version = {}
  46. with open(os.path.join(_here, _this_package, 'version.py')) as f:
  47. exec(f.read(), version)
  48. setup(
  49. name=_this_package,
  50. version=version['__version__'],
  51. description='Description in here.',
  52. long_description=long_description,
  53. author='Bruce Wayne',
  54. author_email='bruce.wayne@example.com',
  55. url='http://example.com',
  56. license='MIT',
  57. packages=[_this_package],
  58. include_package_data=True,
  59. classifiers=[
  60. 'Development Status :: 3 - Alpha',
  61. 'Intended Audience :: Science/Research',
  62. 'Programming Language :: Python :: 2.7',
  63. 'Programming Language :: Python :: 3.6'
  64. ],
  65. cmdclass={'build': extend_build()})

account子目录中放置一个__init__.py脚本:

  1. from .version import __version__
  2. from .account import Account
  3. __all__ = [
  4. '__version__',
  5. 'Account',
  6. ]

再放一个version.py脚本:

  1. __version__ = '0.0.0'

项目的文件结构如下:

  1. .
  2. ├── account
  3. ├── account.cpp
  4. ├── account.hpp
  5. ├── CMakeLists.txt
  6. ├── __init__.py
  7. ├── test.py
  8. └── version.py
  9. ├── CMakeLists.txt
  10. ├── MANIFEST.in
  11. ├── README.rst
  12. └── setup.py

具体实施

本示例基于第9章第5节项目的基础上。

首先,修改account/CMakeLists.txt,添加安装目标:

  1. install(
  2. TARGETS
  3. account
  4. LIBRARY
  5. DESTINATION account
  6. )

安装目标时,README.rst, MANIFEST.insetup.py__init__.pyversion.py将放置在对应的位置上,我们准备使用pybind11测试安装过程:

  1. 为此,在某处创建一个新目录,我们将在那里测试安装。

  2. 在创建的目录中,从本地路径运行pipenv install。调整本地路径,指向setup.py的目录:

    1. $ pipenv install /path/to/cxx-example
  3. 在Pipenv环境中打开一个Python shell:

    1. $ pipenv run python
  4. Python shell中,可以测试我们的CMake包:

    1. >>> from account import Account
    2. >>> account1 = Account()
    3. >>> account1.deposit(100.0)
    4. >>> account1.deposit(100.0)
    5. >>> account1.withdraw(50.0)
    6. >>> print(account1.get_balance())
    7. 150.0

工作原理

${CMAKE_CURRENT_BINARY_DIR}目录包含编译后的account.cpython-36m-x86_64-linux-gnu.so,这个动态库就是使用pybind11构建Python模块。但是请注意,它的名称取决于操作系统(本例中是64位Linux)和Python环境(本例中是Python 3.6)。setup.pys脚本将运行CMake,并根据所选的Python环境(系统Python,Pipenv或虚拟环境)将Python模块安装到正确的路径下。

不过,在安装模块时面临两个挑战:

  • 名称可变
  • CMake外部设置路径

可以使用下面的安装目标来解决这个问题,将在setup.py中定义安装目标位置:

  1. install(
  2. TARGETS
  3. account
  4. LIBRARY
  5. DESTINATION account
  6. )

指示CMake将编译好的Python模块文件安装到相对于安装目标位置的account子目录中(第10章中详细讨论了如何设置目标位置)。setup.py将通过设置CMAKE_INSTALL_PREFIX来设置安装位置,并根据Python环境指向正确的路径。

让我们看看setup.py如何实现的。自下而上来看一下脚本:

  1. setup(
  2. name=_this_package,
  3. version=version['__version__'],
  4. description='Description in here.',
  5. long_description=long_description,
  6. author='Bruce Wayne',
  7. author_email='bruce.wayne@example.com',
  8. url='http://example.com',
  9. license='MIT',
  10. packages=[_this_package],
  11. include_package_data=True,
  12. classifiers=[
  13. 'Development Status :: 3 - Alpha',
  14. 'Intended Audience :: Science/Research',
  15. 'Programming Language :: Python :: 2.7',
  16. 'Programming Language :: Python :: 3.6'
  17. ],
  18. cmdclass={'build': extend_build()})

该脚本包含许多占位符,还包含一些自解释的语句。这里我们将重点介绍最后一个指令cmdclass。这个指令中,通过自定义extend_build函数扩展默认的构建步骤。这个默认的构建步骤如下:

  1. def extend_build():
  2. class build(_build.build):
  3. def run(self):
  4. cwd = os.getcwd()
  5. if spawn.find_executable('cmake') is None:
  6. sys.stderr.write("CMake is required to build this package.\n")
  7. sys.exit(-1)
  8. _source_dir = os.path.split(__file__)[0]
  9. _build_dir = os.path.join(_source_dir, 'build_setup_py')
  10. _prefix = get_python_lib()
  11. try:
  12. cmake_configure_command = [
  13. 'cmake',
  14. '-H{0}'.format(_source_dir),
  15. '-B{0}'.format(_build_dir),
  16. '-DCMAKE_INSTALL_PREFIX={0}'.format(_prefix),
  17. ]
  18. _generator = os.getenv('CMAKE_GENERATOR')
  19. if _generator is not None:
  20. cmake_configure_command.append('-
  21. G{0}'.format(_generator))
  22. spawn.spawn(cmake_configure_command)
  23. spawn.spawn(
  24. ['cmake', '--build', _build_dir, '--target', 'install'])
  25. os.chdir(cwd)
  26. except spawn.DistutilsExecError:
  27. sys.stderr.write("Error while building with CMake\n")
  28. sys.exit(-1)
  29. _build.build.run(self)
  30. return build

首先,检查CMake是否可用。函数执行了两个CMake命令:

  1. cmake_configure_command = [
  2. 'cmake',
  3. '-H{0}'.format(_source_dir),
  4. '-B{0}'.format(_build_dir),
  5. '-DCMAKE_INSTALL_PREFIX={0}'.format(_prefix),
  6. ]
  7. _generator = os.getenv('CMAKE_GENERATOR')
  8. if _generator is not None:
  9. cmake_configure_command.append('-
  10. G{0}'.format(_generator))
  11. spawn.spawn(cmake_configure_command)
  12. spawn.spawn(
  13. ['cmake', '--build', _build_dir, '--target', 'install'])

我们可以设置CMAKE_GENERATOR环境变量来修改生成器。安装目录如下方式设置:

  1. _prefix = get_python_lib()

从安装目录的根目录下,通过distutils.sysconfig导入get_python_lib函数。cmake --build _build_dir --target install命令以一种可移植的方式,构建和安装我们的项目。使用_build_dir而不使用build的原因是,在测试本地安装时,项目可能已经包含了一个build目录,这将与新安装过程发生冲突。对于已经上传到PyPI的包,构建目录的名称并不会带来什么影响。

更多信息

现在我们已经测试了本地安装,准备将包上传到PyPI。在此之前,请确保setup.py中的元数据(例如:项目名称、联系方式和许可协议信息)是合理的,并且项目名称没有与PyPI已存在项目重名。在上传到https://pypi.org 之前,先测试PyPI(https://test.pypi.org )上,进行上载和下载的尝试。

上传之前,我们需要在主目录中创建一个名为.pypirc的文件,其中包含(替换成自己的yourusernameyourpassword):

  1. [distutils]account
  2. index-servers=
  3. pypi
  4. pypitest
  5. [pypi]
  6. username = yourusername
  7. password = yourpassword
  8. [pypitest]
  9. repository = https://test.pypi.org/legacy/
  10. username = yourusername
  11. password = yourpassword

我们将分两步进行。首先,我们在本地创建Release包:

  1. $ python setup.py sdist

第二步中,使用Twine上传生成的分布数据(我们将Twine安装到本地的Pipenv中):

  1. $ pipenv run twine upload dist/* -r pypitest
  2. Uploading distributions to https://test.pypi.org/legacy/
  3. Uploading yourpackage-0.0.0.tar.gz

下一步,从测试实例到,将包安装到一个隔离的环境中:

  1. $ pipenv shell
  2. $ pip install --index-url https://test.pypi.org/simple/ yourpackage

当一切正常,就将我们的包上传到了PyPI:

  1. $ pipenv run twine upload dist/* -r pypi