setup.py

import codecs
import os
import re

from setuptools import setup, find_packages

# Acknowledgement: This setup.py was adapted from Hynek Schlawack's Python
#                  Packaging Guide
# https://hynek.me/articles/sharing-your-labor-of-love-pypi-quick-and-dirty

###################################################################

NAME = "conductor-cli"
PACKAGES = find_packages(where="src")
META_PATH = os.path.join("src", "conductor", "__init__.py")
README_PATH = "README.md"
PYTHON_REQUIRES = ">=3.8"

PACKAGE_DATA = {}
PACKAGE_DIR = {"": "src"}

ENTRY_POINTS = {
    "console_scripts": [
        "cond = conductor.__main__:main",
    ],
}

INSTALL_REQUIRES = [
    "tomli",
]

DEV_REQUIRES = [
    "black",
    "grpcio-tools",
    "mypy",
    "pep517",
    "pylint",
    "pytest",
    "pytest-timeout",
    "pyyaml",
    "setuptools",
    "types-protobuf",
    "types-invoke",
]

ENVS_REQUIRES = [
    "fabric",
    "grpcio",
    "protobuf",
]

EXPLORER_REQUIRES = [
    "fastapi",
    "uvicorn[standard]",
    "pydantic",
]

KEYWORDS = [
    "research",
    "experiment runner",
    "research computing",
    "experiment automation",
    "orchestration",
]

CLASSIFIERS = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: GNU Affero General Public License v3",
    "Programming Language :: Python :: 3 :: Only",
]

###################################################################

HERE = os.path.abspath(os.path.dirname(__file__))


def read(*parts):
    """
    Build an absolute path from *parts* and return the contents of the
    resulting file. Assume UTF-8 encoding.
    """
    with codecs.open(os.path.join(HERE, *parts), "rb", "utf-8") as f:
        return f.read()


META_FILE = read(META_PATH)


def find_meta(meta):
    """
    Extract __*meta*__ from META_FILE.
    """
    meta_match = re.search(
        r"^__{meta}__ = ['\"]([^'\"]*)['\"]".format(meta=meta), META_FILE, re.M
    )
    if meta_match:
        return meta_match.group(1)
    raise RuntimeError("Unable to find __{meta}__ string.".format(meta=meta))


if __name__ == "__main__":
    setup(
        name=NAME,
        description=find_meta("description"),
        version=find_meta("version"),
        author=find_meta("author"),
        author_email=find_meta("email"),
        maintainer=find_meta("author"),
        maintainer_email=find_meta("email"),
        url=find_meta("url"),
        long_description=read(README_PATH),
        long_description_content_type="text/markdown",
        packages=PACKAGES,
        package_data=PACKAGE_DATA,
        package_dir=PACKAGE_DIR,
        python_requires=PYTHON_REQUIRES,
        install_requires=INSTALL_REQUIRES,
        extras_require={
            "all": ENVS_REQUIRES + EXPLORER_REQUIRES,
            "dev": DEV_REQUIRES,
            "envs": ENVS_REQUIRES,
            "explorer": EXPLORER_REQUIRES,
        },
        entry_points=ENTRY_POINTS,
        classifiers=CLASSIFIERS,
        keywords=KEYWORDS,
    )