summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAdam Schwalm <adamschwalm@gmail.com>2021-09-18 04:20:31 (GMT)
committerGitHub <noreply@github.com>2021-09-18 04:20:31 (GMT)
commita6e8db5e8e6780411db749d404715dbe021647c7 (patch)
treedd2b82a52745e2f764788b613fc928670c01f026
parentaf08f1ba40505bf1380c08b57ba4e0b8969a8358 (diff)
downloadcpython-a6e8db5e8e6780411db749d404715dbe021647c7.zip
cpython-a6e8db5e8e6780411db749d404715dbe021647c7.tar.gz
cpython-a6e8db5e8e6780411db749d404715dbe021647c7.tar.bz2
bpo-45235: Fix argparse overrides namespace with subparser defaults (GH-28420)
-rw-r--r--Lib/argparse.py13
-rw-r--r--Lib/test/test_argparse.py6
-rw-r--r--Misc/NEWS.d/next/Library/2021-09-17-16-55-37.bpo-45235.sXnmPA.rst2
3 files changed, 15 insertions, 6 deletions
diff --git a/Lib/argparse.py b/Lib/argparse.py
index d5a0afb..df98999 100644
--- a/Lib/argparse.py
+++ b/Lib/argparse.py
@@ -1209,7 +1209,8 @@ class _SubParsersAction(Action):
# namespace for the relevant parts.
subnamespace, arg_strings = parser.parse_known_args(arg_strings, None)
for key, value in vars(subnamespace).items():
- setattr(namespace, key, value)
+ if not hasattr(namespace, key):
+ setattr(namespace, key, value)
if arg_strings:
vars(namespace).setdefault(_UNRECOGNIZED_ARGS_ATTR, [])
@@ -1843,11 +1844,6 @@ class ArgumentParser(_AttributeHolder, _ActionsContainer):
if action.default is not SUPPRESS:
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
if self.exit_on_error:
try:
@@ -1858,6 +1854,11 @@ class ArgumentParser(_AttributeHolder, _ActionsContainer):
else:
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):
args.extend(getattr(namespace, _UNRECOGNIZED_ARGS_ATTR))
delattr(namespace, _UNRECOGNIZED_ARGS_ATTR)
diff --git a/Lib/test/test_argparse.py b/Lib/test/test_argparse.py
index d369d0f..9926910 100644
--- a/Lib/test/test_argparse.py
+++ b/Lib/test/test_argparse.py
@@ -3079,6 +3079,12 @@ class TestSetDefaults(TestCase):
xparser.set_defaults(foo=2)
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):
parser = ErrorRaisingArgumentParser()
parser.set_defaults(w='W', x='X', y='Y', z='Z')
diff --git a/Misc/NEWS.d/next/Library/2021-09-17-16-55-37.bpo-45235.sXnmPA.rst b/Misc/NEWS.d/next/Library/2021-09-17-16-55-37.bpo-45235.sXnmPA.rst
new file mode 100644
index 0000000..871ec52
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2021-09-17-16-55-37.bpo-45235.sXnmPA.rst
@@ -0,0 +1,2 @@
+Fix an issue where argparse would not preserve values in a provided namespace
+when using a subparser with defaults. \ No newline at end of file