mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-29 05:05:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Closes #12713: Allowed abbreviation of subcommands in argparse.

Browse source
This commit is contained in:
Vinay Sajip 2016-08-18 21:23:48 +01:00
parent 97c1adf393
commit ef948cd058
3 changed files with 95 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

23
Lib/argparse.py
View file

@ -1110,6 +1110,12 @@ class _SubParsersAction(Action):
parser_name = values[0]
arg_strings = values[1:]
# get full parser_name from (optional) abbreviated one
for p in self._name_parser_map:
if p.startswith(parser_name):
parser_name = p
break
# set the parser name if requested
if self.dest is not SUPPRESS:
setattr(namespace, self.dest, parser_name)
@ -2307,11 +2313,18 @@ class ArgumentParser(_AttributeHolder, _ActionsContainer):
def _check_value(self, action, value):
# converted value must be one of the choices (if specified)
if action.choices is not None and value not in action.choices:
args = {'value': value,
'choices': ', '.join(map(repr, action.choices))}
msg = _('invalid choice: %(value)r (choose from %(choices)s)')
raise ArgumentError(action, msg % args)
if action.choices is not None:
ac = [ax for ax in action.choices if str(ax).startswith(str(value))]
if len(ac) == 0:
args = {'value': value,
'choices': ', '.join(map(repr, action.choices))}
msg = _('invalid choice: %(value)r (choose from %(choices)s)')
raise ArgumentError(action, msg % args)
elif len(ac) > 1:
args = {'value': value,
'choices': ', '.join(ac)}
msg = _('ambiguous choice: %(value)r could match %(choices)s')
raise ArgumentError(action, msg % args)
# =======================
# Help-formatting methods