setup.py

import os
import subprocess
import sys
from setuptools import setup, Extension, find_packages
from setuptools.command.build_ext import build_ext


class CMakeExtension(Extension):
    def __init__(self, name):
        Extension.__init__(self, name, sources=[])


class CMakeBuild(build_ext):
    def run(self):
        try:
            subprocess.check_call(['cmake', '--version'])
        except OSError:
            raise RuntimeError("CMake must be installed")

        if not os.path.exists(self.build_temp):
            os.makedirs(self.build_temp)

        for ext in self.extensions:
            self.build_extension(ext)

    def build_extension(self, ext):
        extdir = os.path.abspath(
            os.path.dirname(self.get_ext_fullpath(ext.name)))

        cfg = 'Debug' if self.debug else 'Release'

        target_version = str(sys.version_info.major) + "." + str(
            sys.version_info.minor)
        executable_path = sys.executable

        cmake_args = [
            "-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=" + extdir,
            "-DCMAKE_BUILD_TYPE=" + cfg,
            "-DSEQUENTIAL_LINE_SEARCH_BUILD_COMMAND_DEMOS=OFF",
            "-DSEQUENTIAL_LINE_SEARCH_BUILD_VISUAL_DEMOS=OFF",
            "-DSEQUENTIAL_LINE_SEARCH_BUILD_PHOTO_DEMOS=OFF",
            "-DSEQUENTIAL_LINE_SEARCH_BUILD_PYTHON_BINDING=ON",
            "-DPYBIND11_PYTHON_VERSION=" + target_version,
            "-DPYTHON_EXECUTABLE=" + executable_path,
        ]

        cmake_list_dir = os.path.abspath(os.path.dirname(__file__))

        configure_command = ['cmake', cmake_list_dir] + cmake_args
        build_command = ['cmake', '--build', '.']

        subprocess.check_call(configure_command, cwd=self.build_temp)
        subprocess.check_call(build_command, cwd=self.build_temp)


setup(
    name="pySequentialLineSearch",
    version="0.4",
    author="Yuki Koyama",
    author_email="yuki@koyama.xyz",
    description="Python bindings of sequential line search",
    long_description="",
    long_description_content_type="text/markdown",
    ext_modules=[CMakeExtension("pySequentialLineSearch")],
    url="https://github.com/yuki-koyama/sequential-line-search",
    packages=find_packages(),
    install_requires=["numpy"],
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
    ],
    python_requires=">=3.6",
    cmdclass=dict(build_ext=CMakeBuild),
    zip_safe=False,
)