Skip to content

[3.7] bpo-11874: fix assertion failure in argparse metavar handling (GH-1826) #7527

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion Lib/argparse.py
Original file line number Diff line number Diff line change
Expand Up @@ -327,7 +327,11 @@ def _format_usage(self, usage, actions, groups, prefix):
if len(prefix) + len(usage) > text_width:

# break usage into wrappable parts
part_regexp = r'\(.*?\)+|\[.*?\]+|\S+'
part_regexp = (
r'\(.*?\)+(?=\s|$)|'
r'\[.*?\]+(?=\s|$)|'
r'\S+'
)
opt_usage = format(optionals, groups)
pos_usage = format(positionals, groups)
opt_parts = _re.findall(part_regexp, opt_usage)
Expand Down
42 changes: 33 additions & 9 deletions Lib/test/test_argparse.py
Original file line number Diff line number Diff line change
Expand Up @@ -4955,7 +4955,7 @@ def test_nargs_None_metavar_length0(self):
self.do_test_exception(nargs=None, metavar=tuple())

def test_nargs_None_metavar_length1(self):
self.do_test_no_exception(nargs=None, metavar=("1"))
self.do_test_no_exception(nargs=None, metavar=("1",))

def test_nargs_None_metavar_length2(self):
self.do_test_exception(nargs=None, metavar=("1", "2"))
Expand All @@ -4972,7 +4972,7 @@ def test_nargs_optional_metavar_length0(self):
self.do_test_exception(nargs="?", metavar=tuple())

def test_nargs_optional_metavar_length1(self):
self.do_test_no_exception(nargs="?", metavar=("1"))
self.do_test_no_exception(nargs="?", metavar=("1",))

def test_nargs_optional_metavar_length2(self):
self.do_test_exception(nargs="?", metavar=("1", "2"))
Expand All @@ -4989,7 +4989,7 @@ def test_nargs_zeroormore_metavar_length0(self):
self.do_test_exception(nargs="*", metavar=tuple())

def test_nargs_zeroormore_metavar_length1(self):
self.do_test_no_exception(nargs="*", metavar=("1"))
self.do_test_exception(nargs="*", metavar=("1",))

def test_nargs_zeroormore_metavar_length2(self):
self.do_test_no_exception(nargs="*", metavar=("1", "2"))
Expand All @@ -5006,7 +5006,7 @@ def test_nargs_oneormore_metavar_length0(self):
self.do_test_exception(nargs="+", metavar=tuple())

def test_nargs_oneormore_metavar_length1(self):
self.do_test_no_exception(nargs="+", metavar=("1"))
self.do_test_exception(nargs="+", metavar=("1",))

def test_nargs_oneormore_metavar_length2(self):
self.do_test_no_exception(nargs="+", metavar=("1", "2"))
Expand All @@ -5023,7 +5023,7 @@ def test_nargs_remainder_metavar_length0(self):
self.do_test_no_exception(nargs="...", metavar=tuple())

def test_nargs_remainder_metavar_length1(self):
self.do_test_no_exception(nargs="...", metavar=("1"))
self.do_test_no_exception(nargs="...", metavar=("1",))

def test_nargs_remainder_metavar_length2(self):
self.do_test_no_exception(nargs="...", metavar=("1", "2"))
Expand All @@ -5040,7 +5040,7 @@ def test_nargs_parser_metavar_length0(self):
self.do_test_exception(nargs="A...", metavar=tuple())

def test_nargs_parser_metavar_length1(self):
self.do_test_no_exception(nargs="A...", metavar=("1"))
self.do_test_no_exception(nargs="A...", metavar=("1",))

def test_nargs_parser_metavar_length2(self):
self.do_test_exception(nargs="A...", metavar=("1", "2"))
Expand All @@ -5057,7 +5057,7 @@ def test_nargs_1_metavar_length0(self):
self.do_test_exception(nargs=1, metavar=tuple())

def test_nargs_1_metavar_length1(self):
self.do_test_no_exception(nargs=1, metavar=("1"))
self.do_test_no_exception(nargs=1, metavar=("1",))

def test_nargs_1_metavar_length2(self):
self.do_test_exception(nargs=1, metavar=("1", "2"))
Expand All @@ -5074,7 +5074,7 @@ def test_nargs_2_metavar_length0(self):
self.do_test_exception(nargs=2, metavar=tuple())

def test_nargs_2_metavar_length1(self):
self.do_test_no_exception(nargs=2, metavar=("1"))
self.do_test_exception(nargs=2, metavar=("1",))

def test_nargs_2_metavar_length2(self):
self.do_test_no_exception(nargs=2, metavar=("1", "2"))
Expand All @@ -5091,7 +5091,7 @@ def test_nargs_3_metavar_length0(self):
self.do_test_exception(nargs=3, metavar=tuple())

def test_nargs_3_metavar_length1(self):
self.do_test_no_exception(nargs=3, metavar=("1"))
self.do_test_exception(nargs=3, metavar=("1",))

def test_nargs_3_metavar_length2(self):
self.do_test_exception(nargs=3, metavar=("1", "2"))
Expand All @@ -5118,6 +5118,30 @@ def test_all_exports_everything_but_modules(self):
]
self.assertEqual(sorted(items), sorted(argparse.__all__))


class TestWrappingMetavar(TestCase):

def setUp(self):
self.parser = ErrorRaisingArgumentParser(
'this_is_spammy_prog_with_a_long_name_sorry_about_the_name'
)
# this metavar was triggering library assertion errors due to usage
# message formatting incorrectly splitting on the ] chars within
metavar = '<http[s]://example:1234>'
self.parser.add_argument('--proxy', metavar=metavar)

def test_help_with_metavar(self):
help_text = self.parser.format_help()
self.assertEqual(help_text, textwrap.dedent('''\
usage: this_is_spammy_prog_with_a_long_name_sorry_about_the_name
[-h] [--proxy <http[s]://example:1234>]

optional arguments:
-h, --help show this help message and exit
--proxy <http[s]://example:1234>
'''))


def test_main():
support.run_unittest(__name__)
# Remove global references to avoid looking like we have refleaks.
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
Use a better regex when breaking usage into wrappable parts. Avoids bogus
assertion errors from custom metavar strings.