locust 2.32.6.dev12__py3-none-any.whl → 2.32.6.dev14__py3-none-any.whl

locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.32.6.dev12"
+__version__ = "2.32.6.dev14"
 version = __version__
-__version_tuple__ = (2, 32, 6, "dev12")
+__version_tuple__ = (2, 32, 6, "dev14")
 version_tuple = __version_tuple__

locust/argument_parser.py

@@ -6,6 +6,7 @@ from locust.rpc import Message, zmqrpc
 
 import ast
 import atexit
+import json
 import os
 import platform
 import socket
@@ -28,6 +29,7 @@ import gevent
 import requests
 
 from .util.directory import get_abspaths_in
+from .util.timespan import parse_timespan
 from .util.url import is_url
 
 version = locust.__version__
@@ -369,6 +371,66 @@ def parse_locustfile_option(args=None) -> list[str]:
     return parsed_paths
 
 
+# A hack for setting up an action that raises ArgumentError with configurable error messages.
+# This is meant to be used to immediately block use of deprecated arguments with some helpful messaging.
+
+
+def raise_argument_type_error(err_msg):
+    class ErrorRaisingAction(configargparse.Action):
+        def __call__(self, parser, namespace, values, option_string=None):
+            raise configargparse.ArgumentError(self, err_msg)
+
+    return ErrorRaisingAction
+
+
+# Definitions for some "types" to use with the arguments
+
+
+def timespan(time_str) -> int:
+    try:
+        return parse_timespan(time_str)
+    except ValueError as e:
+        raise configargparse.ArgumentTypeError(str(e))
+
+
+def positive_integer(string) -> int:
+    try:
+        value = int(string)
+    except ValueError:
+        raise configargparse.ArgumentTypeError(f"invalid int value: '{string}'")
+
+    if value < 1:
+        raise configargparse.ArgumentTypeError(
+            f"Invalid --expect-workers argument ({value}), must be a positive number"
+        )
+
+    return value
+
+
+def json_user_config(string):
+    try:
+        if string.endswith(".json"):
+            with open(string) as file:
+                user_config = json.load(file)
+        else:
+            user_config = json.loads(string)
+
+        if not isinstance(user_config, list):
+            user_config = [user_config]
+
+        for config in user_config:
+            if "user_class_name" not in config:
+                raise configargparse.ArgumentTypeError("The user config must specify a user_class_name")
+
+        return user_config
+
+    except json.decoder.JSONDecodeError as e:
+        raise configargparse.ArgumentTypeError(f"The --config-users argument must be a valid JSON string or file: {e}")
+
+    except FileNotFoundError as e:
+        raise configargparse.ArgumentTypeError(str(e))
+
+
 def setup_parser_arguments(parser):
     """
     Setup command-line options
@@ -401,20 +463,13 @@ def setup_parser_arguments(parser):
         help="Rate to spawn users at (users per second). Primarily used together with --headless or --autostart",
         env_var="LOCUST_SPAWN_RATE",
     )
-    parser.add_argument(
-        "--hatch-rate",
-        env_var="LOCUST_HATCH_RATE",
-        metavar="<float>",
-        type=float,
-        default=0,
-        help=configargparse.SUPPRESS,
-    )
     parser.add_argument(
         "-t",
         "--run-time",
         metavar="<time string>",
         help="Stop after the specified amount of time, e.g. (300s, 20m, 3h, 1h30m, etc.). Only used together with --headless or --autostart. Defaults to run forever.",
         env_var="LOCUST_RUN_TIME",
+        type=timespan,
     )
     parser.add_argument(
         "-l",
@@ -425,7 +480,7 @@ def setup_parser_arguments(parser):
     )
     parser.add_argument(
         "--config-users",
-        type=str,
+        type=json_user_config,
         nargs="*",
         help="User configuration as a JSON string or file. A list of arguments or an Array of JSON configuration may be provided",
         env_var="LOCUST_CONFIG_USERS",
@@ -489,6 +544,9 @@ def setup_parser_arguments(parser):
         default=None,
         help=configargparse.SUPPRESS,
         env_var="LOCUST_WEB_AUTH",
+        action=raise_argument_type_error(
+            "The --web-auth parameters has been replaced with --web-login. See https://docs.locust.io/en/stable/extending-locust.html#authentication for details"
+        ),
     )
     web_ui_group.add_argument(
         "--web-login",
@@ -528,7 +586,8 @@ def setup_parser_arguments(parser):
     web_ui_group.add_argument(
         "--legacy-ui",
         default=False,
-        action="store_true",
+        action=raise_argument_type_error("--legacy-ui is no longer supported, remove the parameter to continue"),
+        nargs=0,
         help=configargparse.SUPPRESS,
         env_var="LOCUST_LEGACY_UI",
     )
@@ -561,7 +620,7 @@ def setup_parser_arguments(parser):
     )
     master_group.add_argument(
         "--expect-workers",
-        type=int,
+        type=positive_integer,
         metavar="<int>",
         default=1,
         help="Delay starting the test until this number of workers have connected (only used in combination with --headless/--autostart).",
@@ -584,7 +643,8 @@ def setup_parser_arguments(parser):
     )
     master_group.add_argument(
         "--expect-slaves",
-        action="store_true",
+        action=raise_argument_type_error("The --expect-slaves parameter has been renamed --expect-workers"),
+        nargs=0,
         help=configargparse.SUPPRESS,
     )
 
@@ -608,7 +668,8 @@ Typically ONLY these options (and --locustfile) need to be specified on workers,
     )
     worker_group.add_argument(
         "--slave",
-        action="store_true",
+        action=raise_argument_type_error("The --slave parameter has been renamed --worker"),
+        nargs=0,
         help=configargparse.SUPPRESS,
     )
     worker_group.add_argument(
@@ -720,6 +781,8 @@ Typically ONLY these options (and --locustfile) need to be specified on workers,
         help="Choose between DEBUG/INFO/WARNING/ERROR/CRITICAL. Default is INFO.",
         metavar="<level>",
         env_var="LOCUST_LOGLEVEL",
+        choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
+        type=str.upper,
     )
     log_group.add_argument(
         "--logfile",
@@ -764,6 +827,7 @@ Typically ONLY these options (and --locustfile) need to be specified on workers,
         default="0",
         help="Number of seconds to wait for a simulated user to complete any executing task before exiting. Default is to terminate immediately. When running distributed, this only needs to be specified on the master.",
         env_var="LOCUST_STOP_TIMEOUT",
+        type=timespan,
     )
     other_group.add_argument(
         "--equal-weights",

locust/main.py

@@ -7,6 +7,7 @@ import errno
 import gc
 import importlib.metadata
 import inspect
+import itertools
 import json
 import logging
 import os
@@ -35,7 +36,6 @@ from .stats import (
 )
 from .user.inspectuser import print_task_ratio, print_task_ratio_json
 from .util.load_locustfile import load_locustfile
-from .util.timespan import parse_timespan
 
 # import external plugins if  installed to allow for registering custom arguments etc
 try:
@@ -172,35 +172,13 @@ def main():
     if options.headful:
         options.headless = False
 
-    if options.slave or options.expect_slaves:
-        sys.stderr.write("The --slave/--expect-slaves parameters have been renamed --worker/--expect-workers\n")
-        sys.exit(1)
-
-    if options.web_auth:
-        sys.stderr.write(
-            "The --web-auth parameters has been replaced with --web-login. See https://docs.locust.io/en/stable/extending-locust.html#authentication for details\n"
-        )
-        sys.exit(1)
-
     if options.autoquit != -1 and not options.autostart:
         sys.stderr.write("--autoquit is only meaningful in combination with --autostart\n")
         sys.exit(1)
 
-    if options.hatch_rate:
-        sys.stderr.write("--hatch-rate parameter has been renamed --spawn-rate\n")
-        sys.exit(1)
-
-    if options.legacy_ui:
-        sys.stderr.write("--legacy-ui is no longer supported, remove the parameter to continue\n")
-        sys.exit(1)
-
     # setup logging
     if not options.skip_log_setup:
-        if options.loglevel.upper() in ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]:
-            setup_logging(options.loglevel, options.logfile)
-        else:
-            sys.stderr.write("Invalid --loglevel. Valid values are: DEBUG/INFO/WARNING/ERROR/CRITICAL\n")
-            sys.exit(1)
+        setup_logging(options.loglevel, options.logfile)
 
     children = []
     logger = logging.getLogger(__name__)
@@ -312,13 +290,6 @@ def main():
     if sys.version_info < (3, 10):
         logger.warning("Python 3.9 support is deprecated and will be removed soon")
 
-    if options.stop_timeout:
-        try:
-            options.stop_timeout = parse_timespan(options.stop_timeout)
-        except ValueError:
-            logger.error("Valid --stop-timeout formats are: 20, 20s, 3m, 2h, 1h20m, 3h30m10s, etc.")
-            sys.exit(1)
-
     if options.list_commands:
         print("Available Users:")
         for name in user_classes:
@@ -382,39 +353,17 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
     )
 
     if options.config_users:
-        for json_user_config in options.config_users:
-            try:
-                if json_user_config.endswith(".json"):
-                    with open(json_user_config) as file:
-                        user_config = json.load(file)
-                else:
-                    user_config = json.loads(json_user_config)
-
-                def ensure_user_class_name(config):
-                    if "user_class_name" not in config:
-                        logger.error("The user config must specify a user_class_name")
-                        sys.exit(-1)
-
-                if isinstance(user_config, list):
-                    for config in user_config:
-                        ensure_user_class_name(config)
-
-                        environment.update_user_class(config)
-                else:
-                    ensure_user_class_name(user_config)
-
-                    environment.update_user_class(user_config)
-            except json.decoder.JSONDecodeError as e:
-                logger.error(f"The --config-users argument must be in valid JSON string or file: {e}")
-                sys.exit(-1)
-            except KeyError as e:
-                logger.error(
-                    f"Error applying user config, probably you tried to specify config for a User not present in your locustfile: {e}"
-                )
-                sys.exit(-1)
-            except Exception as e:
-                logger.exception(e)
-                sys.exit(-1)
+        try:
+            for user_config in itertools.chain(*options.config_users):
+                environment.update_user_class(user_config)
+        except KeyError as e:
+            logger.error(
+                f"Error applying user config, probably you tried to specify config for a User not present in your locustfile: {e}"
+            )
+            sys.exit(-1)
+        except Exception as e:
+            logger.exception(e)
+            sys.exit(-1)
 
     if (
         shape_class
@@ -469,13 +418,6 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
     if options.run_time:
         if options.worker:
             logger.info("--run-time specified for a worker node will be ignored.")
-        try:
-            options.run_time = parse_timespan(options.run_time)
-        except ValueError:
-            logger.error(
-                f"Invalid --run-time argument ({options.run_time}), accepted formats are for example 120, 120s, 2m, 3h, 3h30m10s."
-            )
-            sys.exit(1)
 
     if options.csv_prefix:
         base_csv_file = os.path.basename(options.csv_prefix)

{locust-2.32.6.dev12.dist-info → locust-2.32.6.dev14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.32.6.dev12
+Version: 2.32.6.dev14
 Summary: Developer-friendly load testing framework
 Home-page: https://locust.io/
 License: MIT

{locust-2.32.6.dev12.dist-info → locust-2.32.6.dev14.dist-info}/RECORD

@@ -1,7 +1,7 @@
 locust/__init__.py,sha256=Jit8eNUrwuMLqavyFvMZr69e61DILq_KB4yT4MciScw,1681
 locust/__main__.py,sha256=vBQ82334kX06ImDbFlPFgiBRiLIinwNk3z8Khs6hd74,31
-locust/_version.py,sha256=EsuWx7uaLCQYqZs7tBOKGl0mDvqiyrMqLEjEtlkW76A,460
-locust/argument_parser.py,sha256=xB9TQhAALeIZLx0hxoBOVv4dhSNdB15BG0fDYXkyeFc,30137
+locust/_version.py,sha256=X1PvdUAPDrBMpS2DWr-cM7-_WvCLFKVGN5HLKpZOtcA,460
+locust/argument_parser.py,sha256=NsTd3R_7BCLyXsaHi_1ld40GMOVxJ6a3Ju2kyvHGy6o,32486
 locust/clients.py,sha256=XK-xabq2_5GZKMEjebDobvEjeBTtCs8h2EelL7s68Qs,19346
 locust/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 locust/contrib/fasthttp.py,sha256=ehnOrIe1U6C0gW8fRoyHv7-HSSHxfEJxKb5zHw-y1aU,28758
@@ -15,7 +15,7 @@ locust/exception.py,sha256=jGgJ32ubuf4pWdlaVOkbh2Y0LlG0_DHi-lv3ib8ppOE,1791
 locust/html.py,sha256=X9N8u42aHptcQdRZ5cLrBmDuXqRW0yXH_TaWPsWfweY,3691
 locust/input_events.py,sha256=ZIyePyAMuA_YFYWg18g_pE4kwuQV3RbEB250MzXRwjY,3314
 locust/log.py,sha256=5E2ZUOa3V4sfCqa-470Gle1Ik9L5nxYitsjEB9tRwE0,3455
-locust/main.py,sha256=8U2zaGQWvqZMufGd2eny2q8WpSaZFHUEJTqhNNRXF80,30434
+locust/main.py,sha256=O-mHV12OLcufPjJiGMLL-OPK3-SYvuMEPlssF_JHrb8,27913
 locust/py.typed,sha256=gkWLl8yD4mIZnNYYAIRM8g9VarLvWmTAFeUfEbxJLBw,65
 locust/rpc/__init__.py,sha256=nVGoHWFQxZjnhCDWjbgXIbmFbN9sizAjkhvSs9_642c,58
 locust/rpc/protocol.py,sha256=n-rb3GZQcAlldYDj4E4GuFGylYj_26GSS5U29meft5Y,1282
@@ -47,8 +47,8 @@ locust/webui/dist/auth.html,sha256=JYZcdglsfY3B4LEPCteWiNd5kPX21Lx1oaaI00iCTxo,6
 locust/webui/dist/index.html,sha256=uMAhFNZjFQrWASkqHYx0GPsGSs2DO00OrRkvWItpv38,654
 locust/webui/dist/report.html,sha256=wd_TO5cREIfgk0ru49DX0Zlxopb5iWsyuB36v_rfST8,1474524
 poetry.lock,sha256=a4ZAnX6LNBWBJnqtxt4gK0uPSqPsBdDitylFj1rL3LA,214415
-locust-2.32.6.dev12.dist-info/LICENSE,sha256=5hnz-Vpj0Z3kSCQl0LzV2hT1TLc4LHcbpBp3Cy-EuyM,1110
-locust-2.32.6.dev12.dist-info/METADATA,sha256=Cja9wPks95Qb8bqvTOk2oM4PYMiDKI6AsrFyjPooJUw,9991
-locust-2.32.6.dev12.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-locust-2.32.6.dev12.dist-info/entry_points.txt,sha256=RhIxlLwU_Ae_WjimS5COUDLagdCh6IOMyLtgaQxNmlM,43
-locust-2.32.6.dev12.dist-info/RECORD,,
+locust-2.32.6.dev14.dist-info/LICENSE,sha256=5hnz-Vpj0Z3kSCQl0LzV2hT1TLc4LHcbpBp3Cy-EuyM,1110
+locust-2.32.6.dev14.dist-info/METADATA,sha256=x76twu79Sr3C5uUUABKXHgoMKgAv6zaC9mjHyLVXJno,9991
+locust-2.32.6.dev14.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+locust-2.32.6.dev14.dist-info/entry_points.txt,sha256=RhIxlLwU_Ae_WjimS5COUDLagdCh6IOMyLtgaQxNmlM,43
+locust-2.32.6.dev14.dist-info/RECORD,,