# Copyright © 2022 Cerc # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . # Deploys the system components using docker-compose import os from python_on_whales import DockerClient import click from .util import include_exclude_check @click.command() @click.option("--include", help="only start these components") @click.option("--exclude", help="don\'t start these components") @click.option("--cluster", help="specify a non-default cluster name") @click.argument('command') # help: command: up|down|ps @click.pass_context def command(ctx, include, exclude, cluster, command): '''deploy a stack''' # TODO: implement option exclusion and command value constraint lost with the move from argparse to click quiet = ctx.obj.quiet verbose = ctx.obj.verbose dry_run = ctx.obj.dry_run with open("pod-list.txt") as pod_list_file: pods = pod_list_file.read().splitlines() if verbose: print(f'Cluster components: {pods}') # Construct a docker compose command suitable for our purpose compose_files = [] for pod in pods: if include_exclude_check(cluster, include, exclude): compose_file_name = os.path.join("compose", f"docker-compose-{cluster}.yml") compose_files.append(compose_file_name) else: if not quiet: print(f"Excluding: {cluster}") if verbose: print(f"files: {compose_files}") # See: https://gabrieldemarmiesse.github.io/python-on-whales/sub-commands/compose/ docker = DockerClient(compose_files=compose_files) if not dry_run: if command == "up": if verbose: print("Running compose up") docker.compose.up(detach=True) elif command == "down": if verbose: print("Running compose down") docker.compose.down() elif command == "ps": if verbose: print("Running compose ps") docker.compose.ps() elif command == "logs": if verbose: print("Running compose logs") docker.compose.logs()