vantage6 5.0.0a7__py3-none-any.whl → 5.0.0a9__py3-none-any.whl

vantage6/cli/dev/remove.py

@@ -1,94 +0,0 @@
-import subprocess
-import itertools
-from shutil import rmtree
-from pathlib import Path
-
-import click
-
-from vantage6.cli.context.algorithm_store import AlgorithmStoreContext
-from vantage6.common import info
-from vantage6.cli.context.server import ServerContext
-from vantage6.cli.context.node import NodeContext
-from vantage6.cli.server.remove import cli_server_remove
-from vantage6.cli.utils import remove_file
-from vantage6.common.globals import InstanceType
-from vantage6.cli.dev.utils import get_dev_server_context
-
-
-@click.command()
-@click.option("-n", "--name", default=None, help="Name of the configuration.")
-@click.option(
-    "-c",
-    "--config",
-    default=None,
-    help="Path to configuration-file; overrides --name",
-)
-@click.pass_context
-def remove_demo_network(
-    click_ctx: click.Context, name: str | None, config: str | None
-) -> None:
-    """Remove all related demo network files and folders.
-
-    Select a server configuration to remove that server and the nodes attached
-    to it.
-    """
-    ctx = get_dev_server_context(config, name)
-
-    # remove the server
-    for handler in itertools.chain(ctx.log.handlers, ctx.log.root.handlers):
-        handler.close()
-    click_ctx.invoke(cli_server_remove, ctx=ctx, force=True)
-
-    # removing the server import config
-    info("Deleting demo import config file")
-    server_configs = ServerContext.instance_folders(
-        InstanceType.SERVER, ctx.name, system_folders=False
-    )
-    import_config_to_del = Path(server_configs["dev"]) / f"{ctx.name}.yaml"
-    remove_file(import_config_to_del, "import_configuration")
-
-    # also remove the server folder
-    server_configs = ServerContext.instance_folders(
-        InstanceType.SERVER, ctx.name, system_folders=False
-    )
-    server_folder = server_configs["data"]
-    if server_folder.is_dir():
-        rmtree(server_folder)
-
-    # remove the store folder
-    store_configs = AlgorithmStoreContext.instance_folders(
-        InstanceType.ALGORITHM_STORE, f"{ctx.name}_store", system_folders=False
-    )
-    store_folder = store_configs["data"]
-    if store_folder.is_dir():
-        rmtree(store_folder)
-
-    # remove the store config file
-    subprocess.run(
-        [
-            "v6",
-            "algorithm-store",
-            "remove",
-            "-n",
-            f"{ctx.name}_store",
-            "--force",
-            "--user",
-        ]
-    )
-
-    # remove the nodes
-    configs, _ = NodeContext.available_configurations(system_folders=False)
-    node_names = [
-        config.name for config in configs if config.name.startswith(f"{ctx.name}_node_")
-    ]
-    for name in node_names:
-        node_ctx = NodeContext(name, False)
-        for handler in itertools.chain(
-            node_ctx.log.handlers, node_ctx.log.root.handlers
-        ):
-            handler.close()
-        subprocess.run(["v6", "node", "remove", "-n", name, "--user", "--force"])
-
-    # remove data files attached to the network
-    data_dirs_nodes = NodeContext.instance_folders("node", "", False)["dev"]
-    rmtree(Path(data_dirs_nodes / ctx.name))

vantage6/cli/dev/start.py

@@ -1,123 +0,0 @@
-import subprocess
-import click
-
-from vantage6.common import info
-from vantage6.client import Client
-from vantage6.cli.globals import COMMUNITY_STORE
-from vantage6.cli.context.algorithm_store import AlgorithmStoreContext
-from vantage6.cli.context.node import NodeContext
-from vantage6.cli.server.start import cli_server_start
-from vantage6.cli.dev.utils import get_dev_server_context
-from vantage6.common.globals import DEFAULT_API_PATH
-
-
-@click.command()
-@click.option("-n", "--name", default=None, help="Name of the configuration.")
-@click.option(
-    "-c",
-    "--config",
-    default=None,
-    help="Path to configuration-file; overrides --name",
-)
-@click.option(
-    "--server-image", type=str, default=None, help="Server Docker image to use"
-)
-@click.option("--node-image", type=str, default=None, help="Node Docker image to use")
-@click.option(
-    "--store-image", type=str, default=None, help="Algorithm Store Docker image to use"
-)
-@click.pass_context
-def start_demo_network(
-    click_ctx: click.Context,
-    name: str | None,
-    config: str | None,
-    server_image: str | None,
-    node_image: str | None,
-    store_image: str | None,
-) -> None:
-    """Starts running a demo-network.
-
-    Select a server configuration to run its demo network. You should choose a
-    server configuration that you created earlier for a demo network. If you
-    have not created a demo network, you can run `v6 dev create-demo-network` to
-    create one.
-    """
-    ctx = get_dev_server_context(config, name)
-
-    # run the server
-    info("Starting server...")
-    click_ctx.invoke(
-        cli_server_start,
-        ctx=ctx,
-        ip=None,
-        port=None,
-        image=server_image,
-        start_ui=True,
-        ui_port=None,
-        start_rabbitmq=False,
-        rabbitmq_image=None,
-        keep=True,
-        mount_src="",
-        attach=False,
-        system_folders=False,
-    )
-
-    # run the store
-    info("Starting algorithm store...")
-    cmd = ["v6", "algorithm-store", "start", "--name", f"{ctx.name}_store", "--user"]
-    if store_image:
-        cmd.extend(["--image", store_image])
-    subprocess.run(cmd, check=True)
-
-    # run all nodes that belong to this server
-    configs, _ = NodeContext.available_configurations(system_folders=False)
-    node_names = [
-        config.name for config in configs if config.name.startswith(f"{ctx.name}_node_")
-    ]
-    for name in node_names:
-        cmd = ["v6", "node", "start", "--name", name]
-        if node_image:
-            cmd.extend(["--image", node_image])
-        subprocess.run(cmd, check=True)
-
-    # now that both server and store have been started, couple them
-    info("Linking local algorithm store to server...")
-    store_ctxs, _ = AlgorithmStoreContext.available_configurations(system_folders=False)
-    store_ctx = [c for c in store_ctxs if c.name == f"{ctx.name}_store"][0]
-    client = Client(
-        "http://localhost",
-        ctx.config["port"],
-        ctx.config["api_path"],
-        log_level="warn",
-    )
-    # TODO these credentials are hardcoded and may change if changed elsewhere. Link
-    # them together so that they are guaranteed to be the same.
-    USERNAME = "dev_admin"
-    PASSWORD = "password"
-    client.authenticate(USERNAME, PASSWORD)
-    existing_stores = client.store.list().get("data", [])
-    existing_urls = [store["url"] for store in existing_stores]
-    api_path = store_ctx.config.get("api_path", DEFAULT_API_PATH)
-    local_store_url = f"http://localhost:{store_ctx.config['port']}{api_path}"
-    if not local_store_url in existing_urls:
-        client.store.create(
-            algorithm_store_url=local_store_url,
-            name="local store",
-            all_collaborations=True,
-            force=True,  # required to link localhost store
-        )
-        # note that we do not need to register the user as root of the store: this is
-        # already handled in the store config file and is executed on store startup (and
-        # successful because server is already started up at that point)
-    info("Done!")
-
-    # link the community store also to the server
-    info("Linking community algorithm store to local server...")
-    if not COMMUNITY_STORE in existing_urls:
-        client.store.create(
-            algorithm_store_url=COMMUNITY_STORE,
-            name="Community store (read-only)",
-            all_collaborations=True,
-            force=True,  # required to continue when linking localhost server
-        )
-    info("Done!")

vantage6/cli/dev/stop.py

@@ -1,47 +0,0 @@
-import click
-
-from vantage6.cli.context.node import NodeContext
-from vantage6.cli.server.stop import cli_server_stop
-from vantage6.cli.node.stop import cli_node_stop
-from vantage6.cli.algostore.stop import cli_algo_store_stop
-from vantage6.cli.dev.utils import get_dev_server_context
-
-
-@click.command()
-@click.option("-n", "--name", default=None, help="Name of the configuration.")
-@click.option(
-    "-c",
-    "--config",
-    default=None,
-    help="Path to configuration-file; overrides --name",
-)
-@click.pass_context
-def stop_demo_network(
-    click_ctx: click.Context, name: str | None, config: str | None
-) -> None:
-    """Stops a demo network's server and nodes.
-
-    Select a server configuration to stop that server and the nodes attached
-    to it.
-    """
-    ctx = get_dev_server_context(config, name)
-
-    # stop the server
-    click_ctx.invoke(
-        cli_server_stop, name=ctx.name, system_folders=False, all_servers=False
-    )
-
-    # stop the algorithm store
-    click_ctx.invoke(
-        cli_algo_store_stop, name=f"{ctx.name}_store", system_folders=False
-    )
-
-    # stop the nodes
-    configs, _ = NodeContext.available_configurations(False)
-    node_names = [
-        config.name for config in configs if config.name.startswith(f"{ctx.name}_node_")
-    ]
-    for name in node_names:
-        click_ctx.invoke(
-            cli_node_stop, name=name, system_folders=False, all_nodes=False, force=False
-        )

vantage6/cli/dev/utils.py

@@ -1,24 +0,0 @@
-from vantage6.common.globals import InstanceType
-from vantage6.cli.configuration_wizard import select_configuration_questionaire
-from vantage6.cli.context import get_context
-from vantage6.cli.context.server import ServerContext
-
-
-def get_dev_server_context(config: str | None, name: str | None) -> ServerContext:
-    """
-    Get the server context for the development server.
-
-    Parameters
-    ----------
-    config : str | None
-        Path to the configuration file. If None, the name will be used.
-    name : str | None
-        Name of the configuration. If None, a questionaire will be shown.
-    """
-    if config:
-        return ServerContext.from_external_config_file(config)
-    if not name:
-        name = select_configuration_questionaire(
-            InstanceType.SERVER, system_folders=False
-        )
-    return get_context(InstanceType.SERVER, name, system_folders=False)