ucloud-setup/app/host.py

import click
import subprocess
import os
import shutil

from app.helper import (
    clone,
    clone_common,
    clone_etcd_wrapper,
    GitOperation,
    PipenvOperation,
    FileOperation,
)


@click.group()
def host():
    pass


@host.command("setup")
@click.option("--path", required=True)
@click.option("--ssh_username", required=True)
@click.option("--ssh_key_path", required=True, help="For Example, ~/.ssh/id_rsa")
@click.option("--ssh_key_pass", required=True)
@click.option("--etcd_url", required=True)
@click.option("--etcd_password", required=True)
def setup(path, ssh_username, ssh_key_path, ssh_key_pass, etcd_url, etcd_password):
    os.chdir(path)

    repo_name = "ucloud-vm"

    op_result = GitOperation.clone(
        f"https://code.ungleich.ch/ungleich-public/{repo_name}.git"
    )

    content = (
        f"ssh_username={ssh_username}\n"
        f"ssh_pkey={ssh_key_path}\n"
        f"ssh_private_key_password={ssh_key_pass}\n"
        f"ETCD_URL={etcd_url}\n"
        f"ETCD_PASSWORD={etcd_password}\n"
    )

    op_result.add(
        FileOperation.write, path=os.path.join(repo_name, ".env"), content=content
    )

    op_result.add(
        GitOperation.clone,
        path=repo_name,
        url="https://code.ungleich.ch/ungleich-public/ucloud_common",
    )

    op_result.add(
        GitOperation.clone,
        path=repo_name,
        url="https://code.ungleich.ch/ahmedbilal/etcd3_wrapper",
    )

    shutil.copytree(
        src=os.path.join(repo_name, "etcd3_wrapper"),
        dst=os.path.join(repo_name, "ucloud_common", "etcd3_wrapper"),
    )

    op_result.add(PipenvOperation.install, path=repo_name)