dependencies.py

#!/usr/bin/env python3


def setup_requires():
    return [
        "psutil",
        "setuptools",
        "wheel",
        "pytest-runner>=2.9",
        "packaging",
    ]


install_requires = ["packaging", "typer", "rich", "pandas", "matplotlib", "jupytext", *setup_requires()]
install_suggests = {
    "mpi4py": "needed for global data operations",
}
doc_requires = ["sphinx", "matplotlib", "python-slugify", "myst-nb>=0.14.0", "sphinx-autoapi>=1.8", *install_requires]
ci_requires = [
    "pytest",
    "pytest-cov",
    "pytest-xdist",
    "check-manifest",
    "ruff",
    "pre-commit",
    "pytest-pycharm",
    "readme_renderer[md]",
    "rstcheck",
    "codecov",
    "twine",
    "pytest-memprof",
    "pytest-regressions",
    "pytest-datadir",
    "pytest-timeout",
    "docutils",
    "numpy",
]
# these don't go into the pip extras
optional_requirements_file_only = []


def strip_markers(name):
    for m in ";<>=":
        try:
            i = name.index(m)
            name = name[:i].strip()
        except ValueError:
            continue
    return name


def extras():
    import itertools

    import pkg_resources

    def _candidates(blocklist):
        # skip those which aren't needed in our current environment (py ver, platform)
        for pkg in set(itertools.chain(doc_requires, install_suggests.keys())):
            if pkg in blocklist:
                continue
            try:
                marker = next(pkg_resources.parse_requirements(pkg)).marker
                if marker is None or marker.evaluate():
                    yield pkg
            except pkg_resources.RequirementParseError:
                # try to fake a package to get the marker parsed
                stripped = strip_markers(pkg)
                fake_pkg = "pip " + pkg.replace(stripped, "")
                try:
                    marker = next(pkg_resources.parse_requirements(fake_pkg)).marker
                    if marker is None or marker.evaluate():
                        yield pkg
                except pkg_resources.RequirementParseError:
                    continue

    # blocklisted packages need a (minimal) compiler setup
    # - nbresuse, pytest-memprof depend on psutil which has no wheels
    return {
        "ci": ci_requires,
        "docs": doc_requires,
        "full": list(install_suggests.keys()) + ci_requires + doc_requires,
    }


toml_tpl = """
[tool.black]
line-length = 120
skip-string-normalization = true

[build-system]
requires = {0}
build-backend = "setuptools.build_meta"
"""
if __name__ == "__main__":
    note = "# This file is autogenerated. Edit dependencies.py instead"
    print(" ".join([i for i in install_requires + list(install_suggests.keys())]))
    import itertools
    import os

    with open(os.path.join(os.path.dirname(__file__), "requirements.txt"), "w") as req:  # noqa: PTH118, PTH120, PTH123
        req.write(note + "\n")
        for module in sorted(set(itertools.chain(install_requires, setup_requires()))):
            req.write(module + "\n")
    with open(os.path.join(os.path.dirname(__file__), "requirements-optional.txt"), "w") as req:  # noqa: PTH118, PTH120, PTH123
        req.write(note + "\n")
        req.write("-r requirements.txt\n")
        req.write("-r requirements-ci.txt\n")
        for module in sorted(
            set(
                itertools.chain(
                    optional_requirements_file_only,
                    doc_requires,
                    install_suggests.keys(),
                )
            )
        ):
            req.write(module + "\n")
    with open(os.path.join(os.path.dirname(__file__), "requirements-ci.txt"), "w") as req:  # noqa: PTH118, PTH120, PTH123
        req.write("-r requirements.txt\n")
        req.write(note + "\n")
        for module in sorted(ci_requires):
            req.write(module + "\n")
    with open(os.path.join(os.path.dirname(__file__), "pyproject.toml"), "w") as toml:  # noqa: PTH118, PTH120, PTH123
        toml.write(note)
        toml.write(toml_tpl.format(str(setup_requires())))