Gecko [ tsuru-no-tsurugi.tokyo ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2026-07-18 06:45
__init__.py	9.28 KB	-rw-r--r--	2026-04-08 03:24
api.py	2.4 KB	-rw-r--r--	2026-04-08 03:24
attach.py	5.48 KB	-rw-r--r--	2026-04-08 03:24
auto_attach.py	958 B	-rw-r--r--	2026-04-08 03:24
cli_util.py	8.08 KB	-rw-r--r--	2026-04-08 03:24
collect_logs.py	1.73 KB	-rw-r--r--	2026-07-06 22:36
commands.py	3.68 KB	-rw-r--r--	2026-04-08 03:24
config.py	10.72 KB	-rw-r--r--	2026-04-08 03:24
cve.py	6.73 KB	-rw-r--r--	2026-04-08 03:24
cves.py	3.78 KB	-rw-r--r--	2026-04-08 03:24
detach.py	3.38 KB	-rw-r--r--	2026-04-08 03:24
disable.py	10.65 KB	-rw-r--r--	2026-04-08 03:24
enable.py	18.81 KB	-rw-r--r--	2026-04-08 03:24
fix.py	28.63 KB	-rw-r--r--	2026-04-08 03:24
formatter.py	8.75 KB	-rw-r--r--	2026-04-08 03:24
help.py	1.67 KB	-rw-r--r--	2026-04-08 03:24
parser.py	2.72 KB	-rw-r--r--	2026-04-08 03:24
refresh.py	2.42 KB	-rw-r--r--	2026-04-08 03:24
security_status.py	2.79 KB	-rw-r--r--	2026-04-08 03:24
status.py	2.37 KB	-rw-r--r--	2026-04-08 03:24
system.py	1.05 KB	-rw-r--r--	2026-04-08 03:24

Rename

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,
    ):
        entry = HelpEntry(
            position=position, name=name, help_string=help_string
        )
        if entry not in cls.help_entries[category]:
            cls.help_entries[category].append(entry)

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