setup.py

import os

import platform
from setuptools import setup

_SKIP_BUILD_SUGGESTION = """
*******************************************************************************

Building the Rust extension of the library `whenever` failed. See errors above.
Set the `WHENEVER_NO_BUILD_RUST_EXT` environment variable to any value to skip
building the Rust extension and use the (slower) Python version instead.

*******************************************************************************
"""


extra_setup_kwargs = {}

if (
    os.getenv("WHENEVER_NO_BUILD_RUST_EXT")
    or platform.python_implementation() in ("PyPy", "GraalVM")
):
    print("Skipping Whenever Rust extension build")
else:
    from setuptools_rust import Binding, RustExtension, build_rust

    class BuildRust(build_rust):
        def run(self):
            try:
                build_rust.run(self)
            except Exception as e:
                print(_SKIP_BUILD_SUGGESTION)
                raise e

    extra_setup_kwargs.update(
        {
            "rust_extensions": [
                RustExtension(
                    "whenever._whenever",
                    binding=Binding.NoBinding,
                )
            ],
            "cmdclass": {
                "build_rust": BuildRust,
            },
        }
    )

setup(
    **extra_setup_kwargs,
)