build.py

"""Build script."""
from distutils.errors import CCompilerError, DistutilsExecError, DistutilsPlatformError

import numpy
from setuptools import Extension
from setuptools.command.build_ext import build_ext

extensions = [
    Extension(
        "mapanalyzerext",
        sources=["extension/ma_ext.c"],
        include_dirs=[numpy.get_include()],
    ),
]


class BuildFailed(Exception):
    pass


class ExtBuilder(build_ext):
    def run(self):
        try:
            build_ext.run(self)
        except (DistutilsPlatformError, FileNotFoundError):
            pass

    def build_extension(self, ext):
        try:
            build_ext.build_extension(self, ext)
        except (CCompilerError, DistutilsExecError, DistutilsPlatformError, ValueError):
            pass


def build(setup_kwargs):
    setup_kwargs.update(
        {"ext_modules": extensions, "cmdclass": {"build_ext": ExtBuilder}}
    )