Gecko [ oric.aiou.edu.pk ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	9.16 KB	-rw-r--r--
api.py	2.37 KB	-rw-r--r--
attach.py	5.48 KB	-rw-r--r--
auto_attach.py	958 B	-rw-r--r--
cli_util.py	5.93 KB	-rw-r--r--
collect_logs.py	1.3 KB	-rw-r--r--
commands.py	3.68 KB	-rw-r--r--
config.py	9.9 KB	-rw-r--r--
detach.py	3.38 KB	-rw-r--r--
disable.py	10.62 KB	-rw-r--r--
enable.py	17.02 KB	-rw-r--r--
fix.py	28.7 KB	-rw-r--r--
help.py	1.67 KB	-rw-r--r--
parser.py	2.64 KB	-rw-r--r--
refresh.py	2.42 KB	-rw-r--r--
security_status.py	2.79 KB	-rw-r--r--
status.py	2.37 KB	-rw-r--r--
system.py	1.05 KB	-rw-r--r--

Code Editor : parser.py

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

${this.title}

Code Editor : parser.py