locust-cloud 1.21.5.dev2__tar.gz → 1.21.6__tar.gz

{locust_cloud-1.21.5.dev2 → locust_cloud-1.21.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust-cloud
-Version: 1.21.5.dev2
+Version: 1.21.6
 Summary: Locust Cloud
 Project-URL: homepage, https://locust.cloud
 Project-URL: repository, https://github.com/locustcloud/locust-cloud

locust_cloud-1.21.6/locust_cloud/actions.py

@@ -0,0 +1,22 @@
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def delete(session):
+    try:
+        logger.info("Tearing down Locust cloud...")
+        response = session.delete(
+            "/teardown",
+        )
+
+        if response.status_code == 200:
+            logger.debug(f"Response message from teardown: {response.json()['message']}")
+        else:
+            logger.info(
+                f"Could not automatically tear down Locust Cloud: HTTP {response.status_code}/{response.reason} - Response: {response.text} - URL: {response.request.url}"
+            )
+    except KeyboardInterrupt:
+        pass  # don't show nasty callstack
+    except Exception as e:
+        logger.error(f"Could not automatically tear down Locust Cloud: {e.__class__.__name__}:{e}")

{locust_cloud-1.21.5.dev2 → locust_cloud-1.21.6}/locust_cloud/args.py

@@ -8,6 +8,10 @@ import shutil
 import sys
 import tempfile
 
+from locust_cloud.actions import delete
+from locust_cloud.apisession import ApiSession
+from locust_cloud.web_login import logout, web_login
+
 if sys.version_info >= (3, 11):
     import tomllib
 else:
@@ -162,20 +166,42 @@ class MergeToTransferEncodedZipFlat(MergeToTransferEncodedZip):
         setattr(namespace, self.dest, value)
 
 
+class WebLogin(argparse.Action):
+    def __call__(self, parser, namespace, values, option_string=None):
+        web_login()
+        parser.exit()
+
+
+class WebLogout(argparse.Action):
+    def __call__(self, parser, namespace, values, option_string=None):
+        logout()
+        parser.exit()
+
+
+class StackTeardown(argparse.Action):
+    def __call__(self, parser, namespace, values, option_string=None):
+        session = ApiSession(namespace.non_interactive)
+        delete(session)
+        parser.exit()
+
+
 cloud_parser = configargparse.ArgumentParser(add_help=False)
 cloud_parser.add_argument(
     "--login",
-    action="store_true",
+    nargs=0,
+    action=WebLogin,
     help="Launch an interactive session to authenticate your user.\nOnce completed your credentials will be stored and automatically refreshed for quite a long time.\nOnce those expire you will be prompted to perform another login.",
 )
 cloud_parser.add_argument(
     "--logout",
-    action="store_true",
+    nargs=0,
+    action=WebLogout,
     help="Removes the authentication credentials",
 )
 cloud_parser.add_argument(
     "--delete",
-    action="store_true",
+    nargs=0,
+    action=StackTeardown,
     help="Delete a running cluster. Useful if locust-cloud was killed/disconnected or if there was an error.",
 )
 cloud_parser.add_argument(

{locust_cloud-1.21.5.dev2 → locust_cloud-1.21.6}/locust_cloud/cloud.py

@@ -5,6 +5,7 @@ import webbrowser
 from threading import Thread
 
 import requests
+from locust_cloud.actions import delete
 from locust_cloud.apisession import ApiSession
 from locust_cloud.args import combined_cloud_parser
 from locust_cloud.common import __version__
@@ -176,22 +177,3 @@ def main():
     finally:
         logger.debug("Shutting down websocket")
         websocket.shutdown()
-
-
-def delete(session):
-    try:
-        logger.info("Tearing down Locust cloud...")
-        response = session.delete(
-            "/teardown",
-        )
-
-        if response.status_code == 200:
-            logger.debug(f"Response message from teardown: {response.json()['message']}")
-        else:
-            logger.info(
-                f"Could not automatically tear down Locust Cloud: HTTP {response.status_code}/{response.reason} - Response: {response.text} - URL: {response.request.url}"
-            )
-    except KeyboardInterrupt:
-        pass  # don't show nasty callstack
-    except Exception as e:
-        logger.error(f"Could not automatically tear down Locust Cloud: {e.__class__.__name__}:{e}")

{locust_cloud-1.21.5.dev2 → locust_cloud-1.21.6}/locust_cloud/docs/1-first-run.rst

@@ -23,7 +23,7 @@ Then you can launch a distributed test using Locust Cloud:
     [2025-05-13 12:30:58,252] INFO: Deploying (us-east-1, 1.21.5)
     [2025-05-13 12:31:06,366] INFO: Waiting for load generators to be ready...
     [2025-05-13 10:31:11,119] master-lsdhr-8hwhs/INFO/locust.main: Starting Locust 2.37.1 (locust_exporter 1.18.4)
-    [2025-05-13 10:31:11,120] master-lsdhr-8hwhs/INFO/locust.main: Starting web interface at https://us-east-1.webui.locust.cloud/your_customer, press enter to open your default browser.
+    [2025-05-13 10:31:11,120] master-lsdhr-8hwhs/INFO/locust.main: Starting web interface at https://us-east-1.webui.locust.cloud/your_company, press enter to open your default browser.
     [2025-05-13 10:31:11,760] master-lsdhr-8hwhs/INFO/locust.runners: worker-jdnf6-jl8qq_a45c04a8d925448ea647fdcda2e8cf80 (index 0) reported as ready. 1 workers connected.
     [2025-05-13 10:31:11,765] master-lsdhr-8hwhs/INFO/locust.runners: worker-jdnf6-pk8cl_6749cd6c0d244b3a9611d6a4e0a8d30b (index 1) reported as ready. 2 workers connected.
     ...