setup.py

#!/usr/bin/env python3
"""DAPPER benchmarks the performance of data assimilation (DA) methods.

It is usually best to install from source (github),
so that you the code is readily available to play with.
See full README on [github](https://github.com/nansencenter/DAPPER).
"""

import os
import re
import sys

from setuptools import find_packages, setup

# with open("README.md", "r") as fh:
#     long_description = fh.read()
DOCLINES = __doc__.split("\n")

# Dependencies
# Why pin?: https://github.com/nansencenter/DAPPER/issues/41#issuecomment-1381616971
INSTALL_REQUIRES = [
    "scipy>=1.10",
    "numpy~=1.20",
    "jupyter",
    "notebook<7",
    "ipdb",
    "ipython>=5.1",
    "tornado~=6.3",  # 6.2 breaks Jupyter plots (tested on local Mac, Linux)
    "matplotlib~=3.7,<3.9",  # 3.9 breaks plt.ion (tested on M1 Sonoma)
    "mpl-tools==0.2.50",
    "tqdm~=4.31",
    "pyyaml",
    "colorama~=0.4.1",
    "tabulate~=0.8.3",
    "pathos~=0.3",
    "dill==0.3.8",  # Pin vers. to equal GCP.
    "patlib==0.3.5",
    "struct-tools==0.2.5",
    "threadpoolctl>=3.0.0,<4.0.0",
]

EXTRAS = {
    "Qt": ["PyQt5", "qtpy"],
    "debug": ["line_profiler", "pre-commit"],
    "test": [
        "tox",
        "coverage>=5.1",
        "pytest",
        "pytest-cov",
        "pytest-sugar",
        "pytest-benchmark",
        "pytest-clarity",
        "pytest-xdist",
        "pytest-timeout",
    ],
    "lint": ["ruff"],
    "doc": [
        "mkdocs-material",
        "mkdocstrings",
        "mkdocstrings-python",
        "mkdocs-gen-files",
        "mkdocs-literate-nav",
        "mkdocs-section-index",
        "mkdocs-glightbox",
        "mkdocs-jupyter",
        "pybtex",
    ],
    # 'flake8-docstrings', 'flake8-bugbear', 'flake8-comprehensions'],
    "build": ["twine", "jupytext"],
}
EXTRAS["dev"] = (
    EXTRAS["debug"] + EXTRAS["test"] + EXTRAS["lint"] + EXTRAS["build"] + EXTRAS["doc"]
)


def find_version(*file_paths):
    """Get version by regex'ing a file."""
    # Source: packaging.python.org/guides/single-sourcing-package-version

    def read(*parts):
        here = os.path.abspath(os.path.dirname(__file__))
        with open(os.path.join(here, *parts)) as fp:
            return fp.read()

    version_file = read(*file_paths)
    version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", version_file, re.M)
    if version_match:
        return version_match.group(1)
    raise RuntimeError("Unable to find version string.")


# 'python setup.py publish' shortcut.
if sys.argv[-1] == "publish":
    os.system("python setup.py sdist bdist_wheel")
    os.system("twine upload dist/*")
    sys.exit(0)


setup(
    # Basic meta
    name="dapper",
    version=find_version("dapper", "__init__.py"),
    author="Patrick N. Raanes",
    author_email="patrick.n.raanes@gmail.com",
    description=DOCLINES[0],
    long_description="\n".join(DOCLINES[2:]),
    long_description_content_type="text/markdown",
    # >=3.5 for @.
    # >=3.6 for mpl>=3.1.
    # >=3.7 for dataclass, capture_output, dict ordering, np>=1.20.
    # ==3.7 for Colab
    # ==3.9 for the DAPPER/GCP cluster, since dill isn't compat. across versions.
    python_requires=">=3.9",
    install_requires=INSTALL_REQUIRES,
    extras_require=EXTRAS,
    packages=find_packages(),
    # py_modules=["examples.basic_1", "examples.basic_2", "examples.basic_3"],
    package_data={
        "": ["*.txt", "*.md", "*.png", "*.yaml"],
        "dapper.mods.QG.f90": ["*.txt", "*.md", "*.png", "Makefile", "*.f90"],
    },
    # Detailed meta
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Science/Research",
        "Topic :: Scientific/Engineering :: Mathematics",
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    keywords=(
        "data-assimilation enkf kalman-filtering"
        " state-estimation particle-filter kalman"
        " bayesian-methods bayesian-filter chaos"
    ),
    project_urls={
        "Documentation": "https://nansencenter.github.io/DAPPER/",
        "Source": "https://github.com/nansencenter/DAPPER",
        "Tracker": "https://github.com/nansencenter/DAPPER/issues",
        # 'Funding': 'https://donate.pypi.org',
        # 'Say Thanks!': 'http://saythanks.io/to/example',
    },
    # url="https://github.com/nansencenter/DAPPER",
)