setup.py

import subprocess
import sys

from setuptools import Extension


# https://github.com/skvark/opencv-python/blob/master/setup.py
def install_packages(*requirements):
    # No more convenient way until PEP 518 is implemented;
    # setuptools only handles eggs
    subprocess.check_call(
        [sys.executable, "-m", "pip", "install"] + list(requirements)
    )


# https://github.com/skvark/opencv-python/blob/master/setup.py
def get_or_install(name, version=None):
    # Do not import 3rd-party modules into the current process
    import json

    js_packages = json.loads(
        subprocess.check_output(
            [sys.executable, "-m", "pip", "list", "--format", "json"]
        ).decode("ascii")
    )  # valid names & versions are ASCII as per PEP 440
    try:
        [package] = (
            package for package in js_packages if package["name"] == name
        )
    except ValueError:
        install_packages("%s==%s" % (name, version) if version else name)
        return version
    else:
        return package["version"]


def get_long_description():
    with open("README.md") as f:
        long_description = f.read()

    try:
        import github2pypi

        return github2pypi.replace_url(
            slug="wkentaro/octomap-python", content=long_description
        )
    except Exception:
        return long_description


def main():
    get_or_install("cython")
    get_or_install("numpy")
    get_or_install("scikit-build")

    from Cython.Distutils import build_ext
    import numpy
    import skbuild

    ext_modules = [
        Extension(
            "octomap",
            ["octomap/octomap.pyx"],
            include_dirs=[
                "src/octomap/octomap/include",
                "src/octomap/dynamicEDT3D/include",
                numpy.get_include(),
            ],
            library_dirs=[
                "src/octomap/lib",
            ],
            libraries=[
                "dynamicedt3d",
                "octomap",
                "octomath",
            ],
            language="c++",
        )
    ]

    skbuild.setup(
        name="octomap-python",
        version="1.8.0.post12",
        install_requires=["numpy"],
        extras_require={
            "example": ["glooey", "imgviz>=1.2.0", "pyglet", "trimesh[easy]"],
        },
        license="BSD",
        maintainer="Kentaro Wada",
        maintainer_email="www.kentaro.wada@gmail.com",
        url="https://github.com/wkentaro/octomap-python",
        description="Python binding of the OctoMap library.",
        long_description=get_long_description(),
        long_description_content_type="text/markdown",
        classifiers=[
            "Development Status :: 5 - Production/Stable",
            "Intended Audience :: Developers",
            "Natural Language :: English",
            "Programming Language :: Python",
            "Programming Language :: Python :: 2",
            "Programming Language :: Python :: 3",
            "Programming Language :: Python :: Implementation :: CPython",
            "Programming Language :: Python :: Implementation :: PyPy",
        ],
        ext_modules=ext_modules,
        cmdclass={"build_ext": build_ext},
        cmake_source_dir="src/octomap",
    )


if __name__ == "__main__":
    main()