summaryrefslogtreecommitdiff
path: root/mp_tool/util.py
blob: 73b8a42938fb78bbae389602fde6469930a40f55 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import argparse
import platform


class UsageAction(argparse._HelpAction):
    def __call__(self, parser, namespace, values, option_string=None):
        formatter = parser._get_formatter()
        formatter.add_usage(parser.usage,
                            parser._actions,
                            parser._mutually_exclusive_groups)

        formatter.start_section(parser._optionals.title)
        formatter.end_section()
        print(formatter.format_help())
        parser.exit(0)


class HelpAction(argparse._HelpAction):
    def __call__(self, parser, namespace, values, option_string=None):
        formatter = parser._get_formatter()
        formatter.add_usage(parser.usage,
                            parser._actions,
                            parser._mutually_exclusive_groups)

        formatter.start_section(parser._optionals.title)
        formatter.add_text(parser._optionals.description)
        formatter.add_arguments(parser._optionals._group_actions)
        formatter.end_section()

        subparsers_actions = [
            action for action in parser._actions
            if isinstance(action, argparse._SubParsersAction)]

        for subparsers_action in subparsers_actions:
            # get all subparsers and print help
            subparsers = subparsers_action.choices
            for subaction in subparsers_action._get_subactions():
                subparser = subparsers[subaction.dest]
                usage = formatter._format_actions_usage(subparser._actions, [])
                formatter.start_section("{} {}".format(subaction.dest, usage))
                formatter.add_text(subaction.help)
                formatter.add_arguments(subparser._positionals._group_actions)
                formatter.add_arguments(filter(lambda a: not isinstance(a, argparse._HelpAction),
                                               subparser._optionals._group_actions))
                formatter.end_section()

        print(formatter.format_help())
        parser.exit(0)


def format_subcommands_help(parser: argparse.ArgumentParser):
    subparsers_actions = [
        action for action in parser._actions
        if isinstance(action, argparse._SubParsersAction)]

    formatter = parser._get_formatter()
    formatter.add_usage(parser.usage,
                        parser._actions,
                        parser._mutually_exclusive_groups)

    formatter.start_section("Choose subcommand")
    for subparsers_action in subparsers_actions:
        formatter.add_argument(subparsers_action)
    formatter.end_section()

    return formatter.format_help()


def get_default_serial_port() -> str:
    if platform.system() == "Windows":
        return 'COM3'
    else:
        return '/dev/ttyUSB0'