mirror of
https://gitlab.com/kupfer/kupferbootstrap.git
synced 2025-02-23 05:35:44 -05:00
90 lines
2.8 KiB
Python
Executable file
90 lines
2.8 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import click
|
|
import subprocess
|
|
|
|
from os import isatty
|
|
from traceback import format_exc, format_exception_only, format_tb
|
|
from typing import Optional
|
|
|
|
from logger import color_option, logging, quiet_option, setup_logging, verbose_option
|
|
from wrapper import get_wrapper_type, enforce_wrap, nowrapper_option
|
|
from progressbar import progress_bars_option
|
|
|
|
from config.cli import config, config_option, cmd_config
|
|
from packages.cli import cmd_packages
|
|
from flavours.cli import cmd_flavours
|
|
from devices.cli import cmd_devices
|
|
from net.cli import cmd_net
|
|
from chroot.cli import cmd_chroot
|
|
from cache.cli import cmd_cache
|
|
from image.cli import cmd_image
|
|
|
|
|
|
@click.group()
|
|
@click.option('--error-shell', '-E', 'error_shell', is_flag=True, default=False, help='Spawn shell after error occurs')
|
|
@verbose_option
|
|
@quiet_option
|
|
@config_option
|
|
@nowrapper_option
|
|
@color_option
|
|
@progress_bars_option
|
|
def cli(
|
|
verbose: bool = False,
|
|
quiet: bool = False,
|
|
config_file: Optional[str] = None,
|
|
wrapper_override: Optional[bool] = None,
|
|
error_shell: bool = False,
|
|
force_colors: Optional[bool] = None,
|
|
force_progress_bars: Optional[bool] = None,
|
|
):
|
|
setup_logging(verbose, quiet=quiet, force_colors=force_colors)
|
|
# stdout is fd 1
|
|
config.runtime.colors = isatty(1) if force_colors is None else force_colors
|
|
config.runtime.verbose = verbose
|
|
config.runtime.progress_bars = force_progress_bars
|
|
config.runtime.no_wrap = wrapper_override is False
|
|
config.runtime.error_shell = error_shell
|
|
config.try_load_file(config_file)
|
|
if config.file_state.exception:
|
|
logging.warning(f"Config file couldn't be loaded: {config.file_state.exception}")
|
|
if wrapper_override:
|
|
logging.info(f'Force-wrapping in wrapper-type: "{get_wrapper_type()}"!')
|
|
enforce_wrap()
|
|
|
|
|
|
def main():
|
|
try:
|
|
return cli(prog_name='kupferbootstrap')
|
|
except Exception as ex:
|
|
if config.runtime.verbose:
|
|
msg = format_exc()
|
|
else:
|
|
tb_start = ''.join(format_tb(ex.__traceback__, limit=1)).strip('\n')
|
|
tb_end = ''.join(format_tb(ex.__traceback__, limit=-1)).strip('\n')
|
|
short_tb = [
|
|
'Traceback (most recent call last):',
|
|
tb_start,
|
|
'[...]',
|
|
tb_end,
|
|
format_exception_only(ex)[-1], # type: ignore[arg-type]
|
|
]
|
|
msg = '\n'.join(short_tb)
|
|
logging.fatal('\n' + msg)
|
|
if config.runtime.error_shell:
|
|
logging.info('Starting error shell. Type exit to quit.')
|
|
subprocess.call('/bin/bash')
|
|
exit(1)
|
|
|
|
|
|
cli.add_command(cmd_cache)
|
|
cli.add_command(cmd_chroot)
|
|
cli.add_command(cmd_config)
|
|
cli.add_command(cmd_devices)
|
|
cli.add_command(cmd_flavours)
|
|
cli.add_command(cmd_image)
|
|
cli.add_command(cmd_net)
|
|
cli.add_command(cmd_packages)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|