locust-cloud 1.0.20__py3-none-any.whl → 1.1.0__py3-none-any.whl

locust_cloud/auth.py

@@ -4,8 +4,9 @@ from datetime import UTC, datetime, timedelta
 import requests
 from flask import redirect, request, url_for
 from flask_login import UserMixin, current_user, login_user
+from locust_cloud.constants import DEFAULT_LAMBDA_URL
 
-LAMBDA = "https://deployer.locust.cloud/1"
+LAMBDA = DEFAULT_LAMBDA_URL
 
 
 class AuthUser(UserMixin):
@@ -17,6 +18,9 @@ class AuthUser(UserMixin):
 
 
 def set_credentials(credentials, response):
+    if not credentials.get("cognito_client_id_token"):
+        return response
+
     id_token = credentials["cognito_client_id_token"]
     user_sub_id = credentials["user_sub_id"]
     refresh_token = credentials["refresh_token"]

locust_cloud/cloud.py

@@ -190,8 +190,8 @@ def main() -> None:
             )
             sys.exit(1)
 
-        logger.info(f"Logging you into Locust Cloud ({options.lambda_url}, {options.aws_region_name}, v{__version__})")
-
+        logger.info(f"Logging you into Locust Cloud ({options.aws_region_name}, v{__version__})")
+        logger.debug(f"Lambda url: {options.lambda_url}")
         credential_manager = CredentialManager(
             lambda_url=options.lambda_url,
             access_key=options.aws_access_key_id,
@@ -255,6 +255,7 @@ def main() -> None:
                 {"name": "LOCUST_REQUIREMENTS_URL", "value": requirements_url},
                 {"name": "LOCUST_FLAGS", "value": " ".join(locust_options)},
                 {"name": "LOCUST_WEB_HOST_DISPLAY_NAME", "value": "_____"},
+                {"name": "LOCUST_API_BASE_URL", "value": DEFAULT_LAMBDA_URL},
                 *locust_env_variables,
             ],
             "worker_count": worker_count,

{locust_cloud-1.0.20.dist-info → locust_cloud-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: locust-cloud
-Version: 1.0.20
+Version: 1.1.0
 Summary: Locust Cloud
 Requires-Python: >=3.11
 Requires-Dist: blessed==1.20.0

{locust_cloud-1.0.20.dist-info → locust_cloud-1.1.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 locust_cloud/__init__.py,sha256=51wdjGT0nqS5dVuFVDMhthnPacrctteiCMrwRqaPaQg,1867
-locust_cloud/auth.py,sha256=B68r0Hv2mXm_2caqOhofuqaSPVIdWZb6_4_9g7M7zco,3045
-locust_cloud/cloud.py,sha256=LusCi_y8WXyyjR3-_9OhSzzpuhcgia39fecAFNZ--Q8,14547
+locust_cloud/auth.py,sha256=2fcTIq0tYDckDsZZ7yFTHZn5nG7Ir1dlV1QIu_tp3JA,3164
+locust_cloud/cloud.py,sha256=dsWd66H2-xPX4A5f2lsXUzPaOaFFVQ6VXerNoLwDl4E,14660
 locust_cloud/constants.py,sha256=YGgIEnaOT9cAA8F8a8Z8lOmlI78D2fXiVpP0RF213MA,167
 locust_cloud/credential_manager.py,sha256=JbuS22RTGAmngVB9DFLxHIgaonCSImemMGvtWnxBZRE,5515
 locust_cloud/timescale/exporter.py,sha256=j1K-czMUdOLFt5Sop0PTF6TL52vYdngEEI5OEM3Xl60,11128
@@ -15,7 +15,7 @@ locust_cloud/webui/vite.config.ts,sha256=cqxPMkbwEA3H9mGGbuPulQUhIHCosUqm_1usxzs
 locust_cloud/webui/yarn.lock,sha256=OgV48S7IOMpqldFPL5MFG1fjP8HTcDlFSFzMuRCkGWU,216596
 locust_cloud/webui/dist/index.html,sha256=aWAK1eKa4qBfhMonHNf2sHC6iY9RoTzNdvFcc33Wtp4,664
 locust_cloud/webui/dist/assets/index-BG65KOq7.js,sha256=vKgju6VqTxkI5yFtyDr6zGtb3D7V_AcVCYPDm4SeJos,2795922
-locust_cloud-1.0.20.dist-info/METADATA,sha256=6281q7mD58ZMnErkvivktsnlTmQ0X7_Xfx8-cJX8DUI,2189
-locust_cloud-1.0.20.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-locust_cloud-1.0.20.dist-info/entry_points.txt,sha256=PGyAb4e3aTsGS3N3VGShDl6VzJaXy7QwsEgsLOC7V00,57
-locust_cloud-1.0.20.dist-info/RECORD,,
+locust_cloud-1.1.0.dist-info/METADATA,sha256=8o_Pqe64eIvwPi6kgGnojL0p-bzC6YwCbADxpTEO8kY,2188
+locust_cloud-1.1.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+locust_cloud-1.1.0.dist-info/entry_points.txt,sha256=PGyAb4e3aTsGS3N3VGShDl6VzJaXy7QwsEgsLOC7V00,57
+locust_cloud-1.1.0.dist-info/RECORD,,