| Server IP : 127.0.0.2 / Your IP : 3.137.177.255 Web Server : Apache/2.4.18 (Ubuntu) System : User : www-data ( ) PHP Version : 7.0.33-0ubuntu0.16.04.16 Disable Function : disk_free_space,disk_total_space,diskfreespace,dl,exec,fpaththru,getmyuid,getmypid,highlight_file,ignore_user_abord,leak,listen,link,opcache_get_configuration,opcache_get_status,passthru,pcntl_alarm,pcntl_fork,pcntl_waitpid,pcntl_wait,pcntl_wifexited,pcntl_wifstopped,pcntl_wifsignaled,pcntl_wexitstatus,pcntl_wtermsig,pcntl_wstopsig,pcntl_signal,pcntl_signal_dispatch,pcntl_get_last_error,pcntl_strerror,pcntl_sigprocmask,pcntl_sigwaitinfo,pcntl_sigtimedwait,pcntl_exec,pcntl_getpriority,pcntl_setpriority,php_uname,phpinfo,posix_ctermid,posix_getcwd,posix_getegid,posix_geteuid,posix_getgid,posix_getgrgid,posix_getgrnam,posix_getgroups,posix_getlogin,posix_getpgid,posix_getpgrp,posix_getpid,posix,_getppid,posix_getpwnam,posix_getpwuid,posix_getrlimit,posix_getsid,posix_getuid,posix_isatty,posix_kill,posix_mkfifo,posix_setegid,posix_seteuid,posix_setgid,posix_setpgid,posix_setsid,posix_setuid,posix_times,posix_ttyname,posix_uname,pclose,popen,proc_open,proc_close,proc_get_status,proc_nice,proc_terminate,shell_exec,source,show_source,system,virtual MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/uaclient/cli/ |
Upload File : |
import argparse
from collections import OrderedDict
from enum import Enum
from typing import List, NamedTuple # noqa: F401
from uaclient import messages
HelpEntry = NamedTuple(
"HelpEntry", [("position", int), ("name", str), ("help_string", str)]
)
class HelpCategory(Enum):
class _Value:
def __init__(self, code: str, msg: str):
self.code = code
self.msg = msg
QUICKSTART = _Value("quickstart", messages.CLI_HELP_HEADER_QUICK_START)
SECURITY = _Value("security", messages.CLI_HELP_HEADER_SECURITY)
TROUBLESHOOT = _Value(
"troubleshoot", messages.CLI_HELP_HEADER_TROUBLESHOOT
)
OTHER = _Value("other", messages.CLI_HELP_HEADER_OTHER)
FLAGS = _Value("flags", messages.CLI_FLAGS)
def __str__(self):
return self.value.code
@property
def header(self):
return self.value.msg
class ProArgumentParser(argparse.ArgumentParser):
help_entries = OrderedDict(
[
(HelpCategory.QUICKSTART, []),
(HelpCategory.SECURITY, []),
(HelpCategory.TROUBLESHOOT, []),
(HelpCategory.OTHER, []),
(HelpCategory.FLAGS, []),
]
) # type: OrderedDict[HelpCategory, List[HelpEntry]]
@classmethod
def add_help_entry(
cls,
category: HelpCategory,
name: str,
help_string: str,
position: int = 0,
):
cls.help_entries[category].append(
HelpEntry(position=position, name=name, help_string=help_string)
)
def __init__(self, *args, use_main_help: bool = True, **kwargs):
super().__init__(*args, **kwargs)
self.use_main_help = use_main_help
def print_help_for_command(self, command: str):
args_list = command.split()
args_list.append("--help")
try:
self.parse_args(args_list)
# We want help for any specific command,
# but without exiting right after
except SystemExit:
pass
def format_help(self):
if self.use_main_help:
return super().format_help()
help_output = self.format_usage()
for category, items in self.help_entries.items():
help_output += "\n"
help_output += "{}:".format(category.header)
help_output += "\n"
for item in sorted(items, key=lambda item: item.position):
help_output += "\n"
help_output += " {:<17}{}".format(item.name, item.help_string)
help_output += "\n"
if self.epilog:
help_output += "\n"
help_output += self.epilog
help_output += "\n"
return help_output