locust 2.40.6.dev1__tar.gz → 2.40.6.dev5__tar.gz

{locust-2.40.6.dev1 → locust-2.40.6.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.40.6.dev1
+Version: 2.40.6.dev5
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust

{locust-2.40.6.dev1 → locust-2.40.6.dev5}/locust/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.40.6.dev1'
-__version_tuple__ = version_tuple = (2, 40, 6, 'dev1')
+__version__ = version = '2.40.6.dev5'
+__version_tuple__ = version_tuple = (2, 40, 6, 'dev5')
 
 __commit_id__ = commit_id = None

{locust-2.40.6.dev1 → locust-2.40.6.dev5}/locust/argument_parser.py

@@ -7,6 +7,7 @@ from locust.rpc import Message, zmqrpc
 import argparse
 import ast
 import atexit
+import difflib
 import json
 import os
 import platform
@@ -60,6 +61,15 @@ class LocustArgumentParser(configargparse.ArgumentParser):
     optionally exclude arguments from the UI.
     """
 
+    def error(self, message):
+        if "unrecognized arguments:" in message:
+            bad_arg = message.split("unrecognized arguments:")[1].strip().split()[0]
+            options = [opt for action in self._actions for opt in action.option_strings]
+            suggestion = difflib.get_close_matches(bad_arg, options, n=1)
+            if suggestion:
+                message += f"\nDid you mean '{suggestion[0]}'?"
+        self.exit(2, f"{self.prog}: error: {message}\n")
+
     def add_argument(self, *args, **kwargs) -> configargparse.Action:
         """
         This method supports the same args as ArgumentParser.add_argument(..)