# Copyright © 2022, 2023 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 . import click import os from shutil import copyfile import sys default_spec_file_content = """data_dir: /my/path node_name: my-node-name """ def make_default_deployment_dir(): return "deployment-001" @click.command() @click.option("--output", required=True, help="Write yaml spec file here") @click.pass_context def init(ctx, output): with open(output, "w") as output_file: output_file.write(default_spec_file_content) @click.command() @click.option("--spec-file", required=True, help="Spec file to use to create this deployment") @click.option("--deployment-dir", help="Create deployment files in this directory") @click.pass_context def create(ctx, spec_file, deployment_dir): # TODO: check spec-file exists and is readable if deployment_dir is None: deployment_dir = make_default_deployment_dir() if os.path.exists(deployment_dir): print(f"Error: {deployment_dir} already exists") sys.exit(1) os.mkdir(deployment_dir) # Copy spec file into the deployment dir copyfile(spec_file, os.path.join(deployment_dir, os.path.basename(spec_file)))