%PDF- %PDF-
Direktori : /opt/cloudlinux/venv/lib64/python3.11/site-packages/clwizard/ |
Current File : //opt/cloudlinux/venv/lib64/python3.11/site-packages/clwizard/parser.py |
# coding=utf-8 # # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENCE.TXT # import argparse import json from clwizard.modules import get_supported_modules def validate_json_modules(json_string): try: data = json.loads(json_string) except (TypeError, ValueError) as json_error: raise argparse.ArgumentTypeError("Invalid JSON data passed") from json_error supported_modules = get_supported_modules() # TODO: Validate given json data separately for each module for name, instructions in data.items(): if name not in supported_modules: raise argparse.ArgumentTypeError( f"'{name}' module is not supported at the moment, " f"available keys: {list(supported_modules.keys())}" ) if "options" in list(instructions.keys()): if not set(instructions["options"].keys()).issubset( supported_modules[name].supported_options() ): raise argparse.ArgumentTypeError( f"'{name}' module has incorrect options: {list(instructions['options'].keys())}" ) elif instructions: raise argparse.ArgumentTypeError( f"'{name}' module has incorrect instructions: {list(instructions.keys())}" ) return data def parse_cloudlinux_wizard_opts(argv): """ Arg parser for cloudlinux-wizard utility """ parser = argparse.ArgumentParser( prog="cloudlinux-wizard", description="Utility for installing cloudlinux modules", ) # --json parameter is a convenience for UI calls, we do not process it parser.add_argument("--json", action="store_true", help=argparse.SUPPRESS) subparser = parser.add_subparsers(dest="subparser") install_parser = subparser.add_parser( "install", help="Install a module. The installation is performed in the background by default", ) install_parser.add_argument( "--no-async", help="Install in the foreground", action="store_true" ) install_parser.add_argument( "--force", help="Force the installer to ignore existing installations", action="store_true", ) installer_group = install_parser.add_mutually_exclusive_group(required=True) installer_group.add_argument( "--json-data", help="JSON with modules to be installed", type=validate_json_modules, ) installer_group.add_argument( "--resume", help="Restart the installation process with previously given arguments", action="store_true", ) status_parser = subparser.add_parser( "status", help="Return JSON with statuses of all modules" ) status_parser.add_argument( "--initial", "-i", action="store_true", help="Get the current modules' status" ) finish_parser = subparser.add_parser( "finish", help="Finalize the installation from UI and close the wizard" ) cancel_parser = subparser.add_parser( "cancel", help="Remove a module from the installation queue" ) cancel_parser.add_argument("module", choices=list(get_supported_modules())) opts = parser.parse_args(argv) return opts