From 65454291845ee0496d747f7d8892df5f16db3921 Mon Sep 17 00:00:00 2001 From: Kevin Van Brunt Date: Thu, 23 Apr 2020 15:21:58 -0400 Subject: Updated _set_parser_prog() so future calls to add_parser() will set the correct prog value This makes dynamically adding subcommands after the CLI starts easier. --- cmd2/decorators.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'cmd2') diff --git a/cmd2/decorators.py b/cmd2/decorators.py index 7d097534..d2fdf9c7 100644 --- a/cmd2/decorators.py +++ b/cmd2/decorators.py @@ -87,7 +87,7 @@ def _set_parser_prog(parser: argparse.ArgumentParser, prog: str): is a command name and not sys.argv[0]. :param parser: the parser being edited - :param prog: value for the current parsers prog attribute + :param prog: new value for the parser's prog attribute """ # Set the prog value for this parser parser.prog = prog @@ -95,6 +95,10 @@ def _set_parser_prog(parser: argparse.ArgumentParser, prog: str): # Set the prog value for the parser's subcommands for action in parser._actions: if isinstance(action, argparse._SubParsersAction): + # Set the _SubParsersAction's _prog_prefix value. That way if its add_parser() method is called later, + # the correct prog value will be set on the parser being added. + action._prog_prefix = parser.prog + # The keys of action.choices are subcommand names as well as subcommand aliases. The aliases point to the # same parser as the actual subcommand. We want to avoid placing an alias into a parser's prog value. # Unfortunately there is nothing about an action.choices entry which tells us it's an alias. In most cases -- cgit v1.2.1