Skip to content

Commit 6de1754

Browse files
serhiy-storchakamiss-islington
authored andcommitted
pythongh-81691: Fix handling of multiple "--" (double dashes) in argparse (pythonGH-124233)
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 112b170 commit 6de1754

3 files changed

Lines changed: 67 additions & 8 deletions

File tree

Lib/argparse.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2098,6 +2098,11 @@ def consume_positionals(start_index):
20982098
# and add the Positional and its args to the list
20992099
for action, arg_count in zip(positionals, arg_counts):
21002100
args = arg_strings[start_index: start_index + arg_count]
2101+
# Strip out the first '--' if it is not in PARSER or REMAINDER arg.
2102+
if (action.nargs not in [PARSER, REMAINDER]
2103+
and arg_strings_pattern.find('-', start_index,
2104+
start_index + arg_count) >= 0):
2105+
args.remove('--')
21012106
start_index += arg_count
21022107
if args and action.deprecated and action.dest not in warned:
21032108
self._warning(_("argument '%(argument_name)s' is deprecated") %
@@ -2498,13 +2503,6 @@ def parse_known_intermixed_args(self, args=None, namespace=None):
24982503
# Value conversion methods
24992504
# ========================
25002505
def _get_values(self, action, arg_strings):
2501-
# for everything but PARSER, REMAINDER args, strip out first '--'
2502-
if not action.option_strings and action.nargs not in [PARSER, REMAINDER]:
2503-
try:
2504-
arg_strings.remove('--')
2505-
except ValueError:
2506-
pass
2507-
25082506
# optional argument produces a default when not present
25092507
if not arg_strings and action.nargs == OPTIONAL:
25102508
if action.option_strings:

Lib/test/test_argparse.py

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

57465746
def test_double_dash(self):
5747-
parser = argparse.ArgumentParser()
5747+
parser = argparse.ArgumentParser(exit_on_error=False)
5748+
parser.add_argument('-f', '--foo')
5749+
parser.add_argument('bar', nargs='*')
5750+
5751+
args = parser.parse_args(['--foo=--'])
5752+
self.assertEqual(NS(foo='--', bar=[]), args)
5753+
self.assertRaisesRegex(argparse.ArgumentError,
5754+
'argument -f/--foo: expected one argument',
5755+
parser.parse_args, ['--foo', '--'])
5756+
args = parser.parse_args(['-f--'])
5757+
self.assertEqual(NS(foo='--', bar=[]), args)
5758+
self.assertRaisesRegex(argparse.ArgumentError,
5759+
'argument -f/--foo: expected one argument',
5760+
parser.parse_args, ['-f', '--'])
5761+
args = parser.parse_args(['--foo', 'a', '--', 'b', 'c'])
5762+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5763+
args = parser.parse_args(['a', 'b', '--foo', 'c'])
5764+
self.assertEqual(NS(foo='c', bar=['a', 'b']), args)
5765+
args = parser.parse_args(['a', '--', 'b', '--foo', 'c'])
5766+
self.assertEqual(NS(foo=None, bar=['a', 'b', '--foo', 'c']), args)
5767+
args = parser.parse_args(['a', '--', 'b', '--', 'c', '--foo', 'd'])
5768+
self.assertEqual(NS(foo=None, bar=['a', 'b', '--', 'c', '--foo', 'd']), args)
5769+
5770+
parser = argparse.ArgumentParser(exit_on_error=False)
57485771
parser.add_argument('-f', '--foo', nargs='*')
57495772
parser.add_argument('bar', nargs='*')
57505773

@@ -5758,6 +5781,41 @@ def test_double_dash(self):
57585781
self.assertEqual(NS(foo=[], bar=[]), args)
57595782
args = parser.parse_args(['--foo', 'a', 'b', '--', 'c', 'd'])
57605783
self.assertEqual(NS(foo=['a', 'b'], bar=['c', 'd']), args)
5784+
args = parser.parse_args(['a', 'b', '--foo', 'c', 'd'])
5785+
self.assertEqual(NS(foo=['c', 'd'], bar=['a', 'b']), args)
5786+
args = parser.parse_args(['a', '--', 'b', '--foo', 'c', 'd'])
5787+
self.assertEqual(NS(foo=None, bar=['a', 'b', '--foo', 'c', 'd']), args)
5788+
args, argv = parser.parse_known_args(['a', 'b', '--foo', 'c', '--', 'd'])
5789+
self.assertEqual(NS(foo=['c'], bar=['a', 'b']), args)
5790+
self.assertEqual(argv, ['--', 'd'])
5791+
5792+
parser = argparse.ArgumentParser(exit_on_error=False)
5793+
parser.add_argument('foo')
5794+
parser.add_argument('bar', nargs='*')
5795+
5796+
args = parser.parse_args(['--', 'a', 'b', 'c'])
5797+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5798+
args = parser.parse_args(['a', '--', 'b', 'c'])
5799+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5800+
args = parser.parse_args(['a', 'b', '--', 'c'])
5801+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5802+
args = parser.parse_args(['a', '--', 'b', '--', 'c'])
5803+
self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args)
5804+
args = parser.parse_args(['--', '--', 'a', '--', 'b', 'c'])
5805+
self.assertEqual(NS(foo='--', bar=['a', '--', 'b', 'c']), args)
5806+
5807+
parser = argparse.ArgumentParser(exit_on_error=False)
5808+
parser.add_argument('foo')
5809+
parser.add_argument('bar', nargs=argparse.REMAINDER)
5810+
5811+
args = parser.parse_args(['--', 'a', 'b', 'c'])
5812+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5813+
args = parser.parse_args(['a', '--', 'b', 'c'])
5814+
self.assertEqual(NS(foo='a', bar=['b', 'c']), args)
5815+
args = parser.parse_args(['a', 'b', '--', 'c'])
5816+
self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args)
5817+
args = parser.parse_args(['a', '--', 'b', '--', 'c'])
5818+
self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args)
57615819

57625820

57635821
# ===========================
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)