Skip to content

Commit 6c6b044

Browse files
[3.12] gh-81691: Fix handling of multiple "--" (double dashes) in argparse (GH-124233) (GH-124267)
Only the first one has now been removed, all subsequent ones are now taken literally. (cherry picked from commit aae1267) Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
1 parent eca08d3 commit 6c6b044

File tree

3 files changed

+67
-8
lines changed

3 files changed

+67
-8
lines changed

Lib/argparse.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2108,6 +2108,11 @@ def consume_positionals(start_index):
21082108
# and add the Positional and its args to the list
21092109
for action, arg_count in zip(positionals, arg_counts):
21102110
args = arg_strings[start_index: start_index + arg_count]
2111+
# Strip out the first '--' if it is not in PARSER or REMAINDER arg.
2112+
if (action.nargs not in [PARSER, REMAINDER]
2113+
and arg_strings_pattern.find('-', start_index,
2114+
start_index + arg_count) >= 0):
2115+
args.remove('--')
21112116
start_index += arg_count
21122117
take_action(action, args)
21132118

@@ -2503,13 +2508,6 @@ def parse_known_intermixed_args(self, args=None, namespace=None):
25032508
# Value conversion methods
25042509
# ========================
25052510
def _get_values(self, action, arg_strings):
2506-
# for everything but PARSER, REMAINDER args, strip out first '--'
2507-
if not action.option_strings and action.nargs not in [PARSER, REMAINDER]:
2508-
try:
2509-
arg_strings.remove('--')
2510-
except ValueError:
2511-
pass
2512-
25132511
# optional argument produces a default when not present
25142512
if not arg_strings and action.nargs == OPTIONAL:
25152513
if action.option_strings:

Lib/test/test_argparse.py

Lines changed: 59 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5384,7 +5384,30 @@ def test_zero_or_more_optional(self):
53845384
self.assertEqual(NS(x=[]), args)
53855385

53865386
def test_double_dash(self):
5387-
parser = argparse.ArgumentParser()
5387+
parser = argparse.ArgumentParser(exit_on_error=False)
5388+
parser.add_argument('-f', '--foo')
5389+
parser.add_argument('bar', nargs='*')
5390+
5391+
args = parser.parse_args(['--foo=--'])
5392+
self.assertEqual(NS(foo='--', bar=[]), args)
5393+
self.assertRaisesRegex(argparse.ArgumentError,
5394+
'argument -f/--foo: expected one argument',
5395+
parser.parse_args, ['--foo', '--'])
5396+
args = parser.parse_args(['-f--'])
5397+
self.assertEqual(NS(foo='--', bar=[]), args)
5398+
self.assertRaisesRegex(argparse.ArgumentError,
5399+
'argument -f/--foo: expected one argument',
5400+
parser.parse_args, ['-f', '--'])
5401+
args = parser.parse_args(['--foo', 'a', '--', 'b', 'c'])
5402+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5403+
args = parser.parse_args(['a', 'b', '--foo', 'c'])
5404+
self.assertEqual(NS(foo='c', bar=['a', 'b']), args)
5405+
args = parser.parse_args(['a', '--', 'b', '--foo', 'c'])
5406+
self.assertEqual(NS(foo=None, bar=['a', 'b', '--foo', 'c']), args)
5407+
args = parser.parse_args(['a', '--', 'b', '--', 'c', '--foo', 'd'])
5408+
self.assertEqual(NS(foo=None, bar=['a', 'b', '--', 'c', '--foo', 'd']), args)
5409+
5410+
parser = argparse.ArgumentParser(exit_on_error=False)
53885411
parser.add_argument('-f', '--foo', nargs='*')
53895412
parser.add_argument('bar', nargs='*')
53905413

@@ -5398,6 +5421,41 @@ def test_double_dash(self):
53985421
self.assertEqual(NS(foo=[], bar=[]), args)
53995422
args = parser.parse_args(['--foo', 'a', 'b', '--', 'c', 'd'])
54005423
self.assertEqual(NS(foo=['a', 'b'], bar=['c', 'd']), args)
5424+
args = parser.parse_args(['a', 'b', '--foo', 'c', 'd'])
5425+
self.assertEqual(NS(foo=['c', 'd'], bar=['a', 'b']), args)
5426+
args = parser.parse_args(['a', '--', 'b', '--foo', 'c', 'd'])
5427+
self.assertEqual(NS(foo=None, bar=['a', 'b', '--foo', 'c', 'd']), args)
5428+
args, argv = parser.parse_known_args(['a', 'b', '--foo', 'c', '--', 'd'])
5429+
self.assertEqual(NS(foo=['c'], bar=['a', 'b']), args)
5430+
self.assertEqual(argv, ['--', 'd'])
5431+
5432+
parser = argparse.ArgumentParser(exit_on_error=False)
5433+
parser.add_argument('foo')
5434+
parser.add_argument('bar', nargs='*')
5435+
5436+
args = parser.parse_args(['--', 'a', 'b', 'c'])
5437+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5438+
args = parser.parse_args(['a', '--', 'b', 'c'])
5439+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5440+
args = parser.parse_args(['a', 'b', '--', 'c'])
5441+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5442+
args = parser.parse_args(['a', '--', 'b', '--', 'c'])
5443+
self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args)
5444+
args = parser.parse_args(['--', '--', 'a', '--', 'b', 'c'])
5445+
self.assertEqual(NS(foo='--', bar=['a', '--', 'b', 'c']), args)
5446+
5447+
parser = argparse.ArgumentParser(exit_on_error=False)
5448+
parser.add_argument('foo')
5449+
parser.add_argument('bar', nargs=argparse.REMAINDER)
5450+
5451+
args = parser.parse_args(['--', 'a', 'b', 'c'])
5452+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5453+
args = parser.parse_args(['a', '--', 'b', 'c'])
5454+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5455+
args = parser.parse_args(['a', 'b', '--', 'c'])
5456+
self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args)
5457+
args = parser.parse_args(['a', '--', 'b', '--', 'c'])
5458+
self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args)
54015459

54025460

54035461
# ===========================
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
Fix handling of multiple ``"--"`` (double dashes) in :mod:`argparse`. Only
2+
the first one has now been removed, all subsequent ones are now taken
3+
literally.

0 commit comments

Comments
 (0)