mirror of
https://github.com/yavook/kiwi-scp.git
synced 2024-11-23 05:13:01 +00:00
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
import logging
|
|
import os
|
|
from ipaddress import IPv4Network
|
|
from pathlib import Path
|
|
|
|
import click
|
|
|
|
from .cli import KiwiCommandType, KiwiCommand
|
|
from .decorators import kiwi_command
|
|
from .._constants import KIWI_CONF_NAME
|
|
from ..config import KiwiConfig
|
|
from ..instance import Instance
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
@click.option(
|
|
"-d",
|
|
"--directory",
|
|
help=f"initialize a kiwi-scp instance in another directory",
|
|
type=click.Path(
|
|
path_type=Path,
|
|
dir_okay=True,
|
|
writable=True,
|
|
),
|
|
)
|
|
@click.option(
|
|
"-f/-F",
|
|
"--force/--no-force",
|
|
help=f"use default values even if {KIWI_CONF_NAME} is present",
|
|
)
|
|
@kiwi_command(
|
|
short_help="Initializes kiwi-scp",
|
|
)
|
|
class InitCommand(KiwiCommand):
|
|
"""Initialize or reconfigure a kiwi-scp instance"""
|
|
|
|
type = KiwiCommandType.INSTANCE
|
|
|
|
@classmethod
|
|
def run_for_instance(cls, instance: Instance, directory: Path = None, force: bool = None) -> None:
|
|
if directory is not None:
|
|
instance.directory = directory
|
|
|
|
current_config = KiwiConfig() if force else instance.config
|
|
|
|
# check force switch
|
|
if force and os.path.isfile(KIWI_CONF_NAME):
|
|
_logger.warning(f"About to overwrite an existing '{KIWI_CONF_NAME}'!")
|
|
|
|
# build new kiwi dict
|
|
kiwi_dict = current_config.kiwi_dict
|
|
kiwi_dict.update({
|
|
"version": KiwiCommand.user_query("kiwi-scp version to use in this instance", current_config.version),
|
|
"storage": {
|
|
"directory": KiwiCommand.user_query("local directory for service data",
|
|
current_config.storage.directory, Path),
|
|
},
|
|
"network": {
|
|
"name": KiwiCommand.user_query("name for local network hub", current_config.network.name),
|
|
"cidr": KiwiCommand.user_query("CIDRv4 block for local network hub", current_config.network.cidr,
|
|
IPv4Network),
|
|
},
|
|
})
|
|
|
|
# ensure output directory exists
|
|
if not os.path.isdir(instance.directory):
|
|
os.mkdir(instance.directory)
|
|
|
|
# write out the new kiwi.yml
|
|
cfg = KiwiConfig.parse_obj(kiwi_dict)
|
|
with open(instance.directory.joinpath(KIWI_CONF_NAME), "w") as file:
|
|
cfg.dump_kiwi_yml(file)
|