setup.py

#! /usr/bin/env python

import os
import pathlib

import numpy
from Cython.Build import cythonize
from setuptools import Extension
from setuptools import setup


def find_extensions(path="."):
    extensions = pathlib.Path(path).rglob("*.pyx")
    return [
        Extension(
            str(ext.with_suffix("")).replace(os.path.sep, "."),
            [str(ext)],
            extra_compile_args=["-fopenmp"] if "WITH_OPENMP" in os.environ else [],
            extra_link_args=["-fopenmp"] if "WITH_OPENMP" in os.environ else [],
        )
        for ext in extensions
    ]


setup(
    include_dirs=[numpy.get_include()],
    ext_modules=cythonize(
        find_extensions("landlab") + find_extensions("tests"),
        compiler_directives={"embedsignature": True, "language_level": 3},
    ),
)