setup.py

import os
import os.path as op
import platform
import subprocess
import sys

from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext

# Set to 'Release' when building a release
BUILD_TYPE = "Release"


class CMakeExtension(Extension):
    def __init__(self, name, cmake_lists_dir=".", sources=None, **kwa):
        sources = [] if sources is None else sources
        Extension.__init__(self, name, sources=sources, **kwa)
        self.cmake_lists_dir = os.path.abspath(cmake_lists_dir)


class cmake_build_ext(build_ext):
    def build_extensions(self):
        try:
            subprocess.check_output(["cmake", "--version"])
        except OSError:
            raise RuntimeError("Cannot find CMake executable")

        for ext in self.extensions:
            extpath = self.get_ext_fullpath(ext.name)
            extdir = op.abspath(op.dirname(extpath))
            extfile = op.splitext(op.basename(extpath))[0]
            tmpdir = self.build_temp
            cmake_args = [
                "-DBLEND2D_STATIC=TRUE",
                "-DBLEND2DPY_TARGET_NAME={}".format(extfile),
                "-DCMAKE_BUILD_TYPE={}".format(BUILD_TYPE),
                "-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}".format(
                    BUILD_TYPE.upper(), extdir
                ),
                "-DCMAKE_ARCHIVE_OUTPUT_DIRECTORY_{}={}".format(
                    BUILD_TYPE.upper(), extdir
                ),
                "-DPYTHON_EXECUTABLE={}".format(sys.executable),
            ]

            if platform.system() == "Windows":
                plat = "x64" if platform.architecture()[0] == "64bit" else "Win32"
                cmake_args += [
                    "-DCMAKE_WINDOWS_EXPORT_ALL_SYMBOLS=TRUE",
                    "-DCMAKE_RUNTIME_OUTPUT_DIRECTORY_{}={}".format(
                        BUILD_TYPE.upper(), extdir
                    ),
                ]
                if self.compiler.compiler_type == "msvc":
                    cmake_args += [
                        "-DCMAKE_GENERATOR_PLATFORM={}".format(plat),
                    ]
                else:
                    cmake_args += [
                        "-G",
                        "MinGW Makefiles",
                    ]
            elif platform.system() == "Linux":
                cmake_args += [
                    "-DCMAKE_C_FLAGS=-fPIC",
                    "-DCMAKE_CXX_FLAGS=-fPIC",
                ]

            if not op.exists(tmpdir):
                os.makedirs(tmpdir)

            # Config and build the extension
            cmd = ["cmake", ext.cmake_lists_dir] + cmake_args
            subprocess.check_call(cmd, cwd=tmpdir)
            cmd = ["cmake", "--build", ".", "--config", BUILD_TYPE]
            subprocess.check_call(cmd, cwd=tmpdir)


if __name__ == "__main__":
    # See setup.cfg for package metadata
    setup(
        ext_modules=[CMakeExtension("blend2d._capi")],
        cmdclass={"build_ext": cmake_build_ext},
    )