setup.py

"""Build C extensions for sendspin."""

import os

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

_REQUIRE_C_EXT = os.environ.get("SENDSPIN_REQUIRE_C_EXT") == "1"


class OptionalBuildExt(build_ext):
    """A build_ext that treats C extensions as optional.

    If compilation fails (e.g. missing compiler or headers), the error is
    reported but does not abort the build, allowing the numpy fallback to be
    used at runtime.

    Set SENDSPIN_REQUIRE_C_EXT=1 to make compilation failures fatal (used in
    CI to ensure wheels always include the C extension).
    """

    def run(self) -> None:
        try:
            super().run()
        except Exception as exc:
            if _REQUIRE_C_EXT:
                raise
            print(
                "WARNING: Building C extensions for sendspin failed; "
                "falling back to numpy implementation. "
                f"Error: {exc}",
            )

    def build_extension(self, ext: Extension) -> None:
        try:
            super().build_extension(ext)
        except Exception as exc:
            if _REQUIRE_C_EXT:
                raise
            print(
                f"WARNING: Failed to build extension {ext.name!r}; "
                "continuing without it. "
                f"Error: {exc}",
            )


setup(
    ext_modules=[
        Extension("sendspin._volume", sources=["sendspin/_volume.c"]),
    ],
    cmdclass={"build_ext": OptionalBuildExt},
)