setup.py

import sys

from setuptools import find_packages, setup

from dbx import __version__

with open("README.md", "r", encoding="utf-8") as fh:
    long_description = fh.read()

INSTALL_REQUIRES = [
    # to use Databricks and MLflow APIs
    "retry>=0.9.2, <1.0.0",
    "requests>=2.24.0, <3.0.0",
    "mlflow-skinny>=1.28.0,<=2.0.0",
    "databricks-cli>=0.17,<0.18",
    # CLI interface
    "click>=8.1.0,<9.0.0",
    "rich==12.5.1",
    "typer[all]==0.6.1",
    # for templates creation
    "cookiecutter>=1.7.2, <3.0.0",
    # file formats and models
    "pyyaml>=6.0",
    "pydantic>=1.9.1",
    "Jinja2>=2.11.2",
    # misc - enforced to avoid issues with dependent libraries
    "cryptography>=3.3.1,<38.0.0",
    # required by dbx sync
    "aiohttp>=3.8.1",
    "pathspec>=0.9.0",
    "watchdog>=2.1.0",
]

if sys.platform.startswith("win32"):
    INSTALL_REQUIRES.append("pywin32==227")

DEV_REQUIREMENTS = [
    # utilities for documentation
    "mkdocs>=1.1.2,<2.0.0",
    "mkdocs-click>=0.8.0,<1.0",
    "mkdocs-material>=8.4,<9.0.0",
    "mdx-include>=1.4.1,<2.0.0",
    "mkdocs-markdownextradata-plugin>=0.1.7,<0.3.0",
    "mkdocs-glightbox>=0.2.1,<1.0",
    # pre-commit and linting utilities
    "pre-commit>=2.20.0,<3.0.0",
    "prospector==1.7.0",
    "black>=22.3.0,<23.0.0",
    "MarkupSafe>=2.1.1,<3.0.0",
    # testing framework
    "pytest>=7.1.2,<8.0.0",
    "pytest-mock>=3.8.2,<3.9.0",
    "pytest-xdist[psutil]>=2.5.0,<3.0.0",
    "pytest-asyncio>=0.18.3,<1.0.0",
    "pytest-cov>=3.0.0,<4.0.0",
    "pytest-timeout>=2.1.0,<3.0.0",
    "pytest-clarity>=1.0.1,<2.0.0",
    "poetry>=1.2.0",
]

setup(
    name="dbx",
    python_requires=">=3.8",
    packages=find_packages(exclude=["tests", "tests.*"]),
    setup_requires=["wheel>=0.37.1,<0.38"],
    install_requires=INSTALL_REQUIRES,
    extras_require={"dev": DEV_REQUIREMENTS},
    entry_points={"console_scripts": ["dbx=dbx.cli:entrypoint"]},
    long_description=long_description,
    long_description_content_type="text/markdown",
    include_package_data=True,
    version=__version__,
    description="DataBricks CLI eXtensions aka dbx",
    author="Thunder Shiviah, Michael Shtelma, Ivan Trusov",
    license="Databricks License",
    classifiers=[
        "Intended Audience :: Developers",
        "Intended Audience :: System Administrators",
    ],
)