locust 2.23.1.dev7__py3-none-any.whl → 2.23.2.dev7__py3-none-any.whl

locust/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.23.1.dev7'
-__version_tuple__ = version_tuple = (2, 23, 1, 'dev7')
+__version__ = version = '2.23.2.dev7'
+__version_tuple__ = version_tuple = (2, 23, 2, 'dev7')

locust/argument_parser.py

@@ -4,6 +4,7 @@ import locust
 from locust import runners
 from locust.rpc import Message, zmqrpc
 
+import ast
 import atexit
 import os
 import platform
@@ -159,13 +160,19 @@ def is_url(url: str) -> bool:
 def download_locustfile_from_url(url: str) -> str:
     try:
         response = requests.get(url)
+        # Check if response is valid python code
+        ast.parse(response.text)
     except requests.exceptions.RequestException as e:
         sys.stderr.write(f"Failed to get locustfile from: {url}. Exception: {e}")
         sys.exit(1)
+    except SyntaxError:
+        sys.stderr.write(f"Failed to get locustfile from: {url}. Response is not valid python code.")
+        sys.exit(1)
 
     with open(os.path.join(tempfile.gettempdir(), url.rsplit("/", 1)[-1]), "w") as locustfile:
         locustfile.write(response.text)
 
+    # Clean up downloaded files on exit
     def exit_handler():
         try:
             os.remove(locustfile.name)
@@ -202,7 +209,7 @@ See documentation for more details, including how to set options using a file or
         "--locustfile",
         metavar="<filename>",
         default="locustfile",
-        help="The Python file or module that contains your test, e.g. 'my_test.py'. Also accepts multiple comma-separated .py files or a package name/directory. Defaults to 'locustfile'.",
+        help="The Python file or module that contains your test, e.g. 'my_test.py'. Accepts multiple comma-separated .py files, a package name/directory or a url to a remote locustfile. Defaults to 'locustfile'.",
         env_var="LOCUST_LOCUSTFILE",
     )
 

locust/main.py

@@ -132,6 +132,10 @@ def main():
         logging.error("stats.PERCENTILES_TO_CHART parameter should be 2 parameters \n")
         sys.exit(1)
 
+    if len(stats.MODERN_UI_PERCENTILES_TO_CHART) > 6:
+        logging.error("stats.MODERN_UI_PERCENTILES_TO_CHART parameter should be a maximum of 6 parameters \n")
+        sys.exit(1)
+
     def is_valid_percentile(parameter):
         try:
             if 0 < float(parameter) < 1: