bec/bec_server/setup.py

import os
import pathlib
import subprocess

from setuptools import setup

current_path = pathlib.Path(__file__).parent.resolve()

__version__ = "0.14.1"


def run_install(setup_args, bec_deps, editable=False):
    """
    Run the setup function with the given arguments. If editable is True, the dependencies are installed in editable mode.
    """
    if editable:
        # check if "[dev]" was requested
        if "dev" in os.environ.get("EXTRAS_REQUIRE", ""):
            suffix = "[dev]"
        else:
            suffix = ""
        setup(**setup_args)
        deps = [f"{current_path}/../{dep[1]}/" for dep in bec_deps]
        for dep in deps:
            subprocess.run(f"pip install -e {dep}{suffix}", shell=True, check=True)
        return

    install_deps = [dep[0] for dep in bec_deps]
    setup_args["install_requires"].extend(install_deps)
    print(setup_args)
    setup(**setup_args)


if __name__ == "__main__":
    import sys

    bec_deps = [
        ("bec_lib", "bec_lib"),
        ("bec_ipython_client", "bec_client"),
        ("bec_scan_server", "scan_server"),
        ("bec_scan_bundler", "scan_bundler"),
        ("bec_file_writer", "file_writer"),
        ("bec_dap", "data_processing"),
        ("bec_device_server", "device_server"),
        ("bec_scihub", "scihub"),
    ]

    setup_args = {
        "entry_points": {"console_scripts": ["bec-server = bec_server:main"]},
        "install_requires": ["libtmux"],
        "version": __version__,
        "extras_require": {
            "dev": [
                "pytest",
                "pytest-random-order",
                "pytest-asyncio",
                "coverage",
                "black",
                "pylint",
            ]
        },
    }

    editable = bool("-e" in sys.argv)
    print(f"editable: {editable}")
    run_install(setup_args, bec_deps, editable=editable)