mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-19 07:00:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

bpo-45235: Fix argparse overrides namespace with subparser defaults (GH-28420)

Browse source
This commit is contained in:
Adam Schwalm 2021-09-17 23:20:31 -05:00 • committed by GitHub
parent af08f1ba40
commit a6e8db5e8e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
Lib/argparse.py
View file

@ -1209,7 +1209,8 @@ class _SubParsersAction(Action):
# namespace for the relevant parts. # namespace for the relevant parts.
subnamespace, arg_strings = parser.parse_known_args(arg_strings, None) subnamespace, arg_strings = parser.parse_known_args(arg_strings, None)
for key, value in vars(subnamespace).items(): for key, value in vars(subnamespace).items():
setattr(namespace, key, value) if not hasattr(namespace, key):
setattr(namespace, key, value)
if arg_strings: if arg_strings:
vars(namespace).setdefault(_UNRECOGNIZED_ARGS_ATTR, []) vars(namespace).setdefault(_UNRECOGNIZED_ARGS_ATTR, [])
@ -1843,11 +1844,6 @@ class ArgumentParser(_AttributeHolder, _ActionsContainer):
if action.default is not SUPPRESS: if action.default is not SUPPRESS:
setattr(namespace, action.dest, action.default) setattr(namespace, action.dest, action.default)
# add any parser defaults that aren't present
for dest in self._defaults:
if not hasattr(namespace, dest):
setattr(namespace, dest, self._defaults[dest])
# parse the arguments and exit if there are any errors # parse the arguments and exit if there are any errors
if self.exit_on_error: if self.exit_on_error:
try: try:
@ -1858,6 +1854,11 @@ class ArgumentParser(_AttributeHolder, _ActionsContainer):
else: else:
namespace, args = self._parse_known_args(args, namespace) namespace, args = self._parse_known_args(args, namespace)
# add any parser defaults that aren't present
for dest in self._defaults:
if not hasattr(namespace, dest):
setattr(namespace, dest, self._defaults[dest])
if hasattr(namespace, _UNRECOGNIZED_ARGS_ATTR): if hasattr(namespace, _UNRECOGNIZED_ARGS_ATTR):
args.extend(getattr(namespace, _UNRECOGNIZED_ARGS_ATTR)) args.extend(getattr(namespace, _UNRECOGNIZED_ARGS_ATTR))
delattr(namespace, _UNRECOGNIZED_ARGS_ATTR) delattr(namespace, _UNRECOGNIZED_ARGS_ATTR)

6
Lib/test/test_argparse.py
View file

@ -3079,6 +3079,12 @@ class TestSetDefaults(TestCase):
xparser.set_defaults(foo=2) xparser.set_defaults(foo=2)
self.assertEqual(NS(foo=2), parser.parse_args(['X'])) self.assertEqual(NS(foo=2), parser.parse_args(['X']))
def test_set_defaults_on_subparser_with_namespace(self):
parser = argparse.ArgumentParser()
xparser = parser.add_subparsers().add_parser('X')
xparser.set_defaults(foo=1)
self.assertEqual(NS(foo=2), parser.parse_args(['X'], NS(foo=2)))
def test_set_defaults_same_as_add_argument(self): def test_set_defaults_same_as_add_argument(self):
parser = ErrorRaisingArgumentParser() parser = ErrorRaisingArgumentParser()
parser.set_defaults(w='W', x='X', y='Y', z='Z') parser.set_defaults(w='W', x='X', y='Y', z='Z')

2
Misc/NEWS.d/next/Library/2021-09-17-16-55-37.bpo-45235.sXnmPA.rst Normal file
View file

@ -0,0 +1,2 @@
Fix an issue where argparse would not preserve values in a provided namespace
when using a subparser with defaults.