locust-cloud 1.5.5__py3-none-any.whl → 1.5.7__py3-none-any.whl

locust_cloud/__init__.py

@@ -21,6 +21,7 @@ PG_PASSWORD = os.environ.get("PG_PASSWORD")
 PG_DATABASE = os.environ.get("PG_DATABASE")
 PG_PORT = os.environ.get("PG_PORT", 5432)
 GRAPH_VIEWER = os.environ.get("GRAPH_VIEWER")
+MAX_USER_COUNT = os.environ.get("MAX_USER_COUNT")
 logger = logging.getLogger(__name__)
 
 
@@ -78,9 +79,6 @@ def on_locust_init(environment: locust.env.Environment, **_args):
     if not (PG_HOST and PG_USER and PG_PASSWORD and PG_DATABASE and PG_PORT):
         return
 
-    if GRAPH_VIEWER:
-        environment.web_ui.template_args["isGraphViewer"] = True
-
     try:
         pool = create_connection_pool(
             pg_user=PG_USER,
@@ -99,5 +97,10 @@ def on_locust_init(environment: locust.env.Environment, **_args):
         Exporter(environment, pool)
 
     if environment.web_ui:
+        environment.web_ui.template_args["maxUserCount"] = MAX_USER_COUNT
+
+        if GRAPH_VIEWER:
+            environment.web_ui.template_args["isGraphViewer"] = True
+
         register_auth(environment)
         register_query(environment, pool)

locust_cloud/auth.py

@@ -7,6 +7,11 @@ import requests
 import werkzeug
 from flask import redirect, request, url_for
 from flask_login import UserMixin, login_user
+from locust_cloud.constants import (
+    DEFAULT_DEPLOYER_URL,
+)
+
+DEPLOYER_URL = os.environ.get("LOCUSTCLOUD_DEPLOYER_URL", DEFAULT_DEPLOYER_URL)
 
 
 class Credentials(TypedDict):
@@ -62,7 +67,7 @@ def register_auth(environment: locust.env.Environment):
 
         try:
             auth_response = requests.post(
-                f"{environment.parsed_options.deployer_url}/auth/login",
+                f"{DEPLOYER_URL}/auth/login",
                 json={"username": username, "password": password},
             )