gh-53780: Ignore the first "--" (double dash) between an option and command in argparse (GH-124275)

This commit is contained in:
Serhiy Storchaka 2024-09-24 10:35:28 +03:00 committed by GitHub
parent 3094cd17b0
commit c578271366
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 23 additions and 4 deletions

View File

@ -2068,11 +2068,15 @@ class ArgumentParser(_AttributeHolder, _ActionsContainer):
# and add the Positional and its args to the list
for action, arg_count in zip(positionals, arg_counts):
args = arg_strings[start_index: start_index + arg_count]
# Strip out the first '--' if it is not in PARSER or REMAINDER arg.
if (action.nargs not in [PARSER, REMAINDER]
and arg_strings_pattern.find('-', start_index,
# Strip out the first '--' if it is not in REMAINDER arg.
if action.nargs == PARSER:
if arg_strings_pattern[start_index] == '-':
assert args[0] == '--'
args.remove('--')
elif action.nargs != REMAINDER:
if (arg_strings_pattern.find('-', start_index,
start_index + arg_count) >= 0):
args.remove('--')
args.remove('--')
start_index += arg_count
if args and action.deprecated and action.dest not in warned:
self._warning(_("argument '%(argument_name)s' is deprecated") %

View File

@ -5984,6 +5984,20 @@ class TestDoubleDash(TestCase):
"invalid choice: '--'",
parser.parse_args, ['--', 'x', '--', 'run', 'a', 'b'])
def test_subparser_after_multiple_argument_option(self):
parser = argparse.ArgumentParser(exit_on_error=False)
parser.add_argument('--foo', nargs='*')
subparsers = parser.add_subparsers()
parser1 = subparsers.add_parser('run')
parser1.add_argument('-f')
parser1.add_argument('bar', nargs='*')
args = parser.parse_args(['--foo', 'x', 'y', '--', 'run', 'a', 'b', '-f', 'c'])
self.assertEqual(NS(foo=['x', 'y'], f='c', bar=['a', 'b']), args)
self.assertRaisesRegex(argparse.ArgumentError,
"invalid choice: '--'",
parser.parse_args, ['--foo', 'x', '--', '--', 'run', 'a', 'b'])
# ===========================
# parse_intermixed_args tests

View File

@ -0,0 +1 @@
:mod:`argparse` now ignores the first ``"--"`` (double dash) between an option and command.