fixed

Author: abdulrafey38

Lib/argparse.py

@@ -84,7 +84,7 @@
     'ZERO_OR_MORE',
 ]
 
-
+import difflib as _difflib
 import os as _os
 import re as _re
 import sys as _sys
@@ -2543,9 +2543,18 @@ def _get_value(self, action, arg_string):
     def _check_value(self, action, value):
         # converted value must be one of the choices (if specified)
         if action.choices is not None and value not in action.choices:
-            args = {'value': value,
-                    'choices': ', '.join(map(repr, action.choices))}
-            msg = _('invalid choice: %(value)r (choose from %(choices)s)')
+            closest_choice = _difflib.get_close_matches(value, action.choices)
+            args = {
+                'value': value,
+                'choices': ', '.join(map(repr, action.choices)),
+            }
+            if closest_choice := closest_choice and closest_choice[0] or None:
+                args['closest'] = closest_choice
+                msg = _('invalid choice: %(value)r, maybe you meant %(closest)r? '
+                        '(choose from %(choices)s)')
+            else:
+                msg = _('invalid choice: %(value)r (choose from %(choices)s)')
+
             raise ArgumentError(action, msg % args)
 
     # =======================

Lib/test/test_argparse.py

@@ -2187,6 +2187,18 @@ def test_required_subparsers_no_destination_error(self):
         )
 
     def test_wrong_argument_subparsers_no_destination_error(self):
+        parser = ErrorRaisingArgumentParser()
+        subparsers = parser.add_subparsers(required=True)
+        subparsers.add_parser('foo')
+        subparsers.add_parser('bar')
+        with self.assertRaises(ArgumentParserError) as excinfo:
+            parser.parse_args(('test',))
+        self.assertRegex(
+            excinfo.exception.stderr,
+            r"error: argument {foo,bar}: invalid choice: 'baz' \(choose from 'foo', 'bar'\)\n$"
+        )
+
+    def test_wrong_argument_subparsers_no_destination_error_with_closest_choice_input(self):
         parser = ErrorRaisingArgumentParser()
         subparsers = parser.add_subparsers(required=True)
         subparsers.add_parser('foo')
@@ -2195,7 +2207,7 @@ def test_wrong_argument_subparsers_no_destination_error(self):
             parser.parse_args(('baz',))
         self.assertRegex(
             excinfo.exception.stderr,
-            r"error: argument {foo,bar}: invalid choice: 'baz' \(choose from 'foo', 'bar'\)\n$",
+            r"error: argument {foo,bar}: invalid choice: 'baz', maybe you meant bar? \(choose from 'foo', 'bar'\)\n$",
         )
 
     def test_optional_subparsers(self):