locust-cloud 1.0.7__py3-none-any.whl → 1.0.9__py3-none-any.whl

locust_cloud/__init__.py

@@ -1,10 +1,11 @@
-from locust_cloud.auth import register_auth
-from locust_cloud.timescale.exporter import Timescale
-
 import os
 
+os.environ["LOCUST_SKIP_MONKEY_PATCH"] = "1"
+
 from locust import events
 from locust.argument_parser import LocustArgumentParser
+from locust_cloud.auth import register_auth
+from locust_cloud.timescale.exporter import Timescale
 
 PG_USER = os.environ.get("PG_USER")
 PG_HOST = os.environ.get("PG_HOST")
@@ -12,16 +13,21 @@ PG_PASSWORD = os.environ.get("PG_PASSWORD")
 PG_DATABASE = os.environ.get("PG_DATABASE")
 PG_PORT = os.environ.get("PG_PORT", 5432)
 
-os.environ["LOCUST_BUILD_PATH"] = os.path.join(os.path.dirname(__file__), "webui/dist")
-
 
 @events.init_command_line_parser.add_listener
 def add_arguments(parser: LocustArgumentParser):
+    if not PG_HOST:
+        parser.add_argument_group(
+            "locust-cloud",
+            "locust-cloud disabled, because PG_HOST was not set - this is normal for local runs",
+        )
+        return
+
+    os.environ["LOCUST_BUILD_PATH"] = os.path.join(os.path.dirname(__file__), "webui/dist")
     locust_cloud = parser.add_argument_group(
         "locust-cloud",
-        "Arguments for use with Locust cloud!",
+        "Arguments for use with Locust cloud",
     )
-
     locust_cloud.add_argument(
         "--exporter",
         default=True,
@@ -40,6 +46,9 @@ def add_arguments(parser: LocustArgumentParser):
 
 @events.init.add_listener
 def on_locust_init(environment, **args):
+    if not PG_HOST:
+        return
+
     if environment.parsed_options.exporter:
         Timescale(
             environment,

locust_cloud/auth.py

@@ -1,5 +1,5 @@
-import datetime
 import os
+from datetime import UTC, datetime, timedelta
 
 import requests
 from flask import redirect, request, url_for
@@ -21,9 +21,9 @@ def set_credentials(credentials, response):
     user_sub_id = credentials["user_sub_id"]
     refresh_token = credentials["refresh_token"]
 
-    response.set_cookie("cognito_token", id_token, expires=datetime.datetime.utcnow() + datetime.timedelta(days=1))
-    response.set_cookie("user_token", refresh_token, expires=datetime.datetime.utcnow() + datetime.timedelta(days=365))
-    response.set_cookie("user_sub_id", user_sub_id, expires=datetime.datetime.utcnow() + datetime.timedelta(days=365))
+    response.set_cookie("cognito_token", id_token, expires=datetime.now(tz=UTC) + timedelta(days=1))
+    response.set_cookie("user_token", refresh_token, expires=datetime.now(tz=UTC) + timedelta(days=365))
+    response.set_cookie("user_sub_id", user_sub_id, expires=datetime.now(tz=UTC) + timedelta(days=365))
 
     return response
 

{locust_cloud-1.0.7.dist-info → locust_cloud-1.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: locust-cloud
-Version: 1.0.7
+Version: 1.0.9
 Summary: Locust Cloud
 Requires-Python: >=3.11
 Requires-Dist: blessed==1.20.0
@@ -20,7 +20,7 @@ Requires-Dist: idna==3.7
 Requires-Dist: inquirer==2.10.1
 Requires-Dist: jmespath==1.0.1
 Requires-Dist: kubernetes==29.0.0
-Requires-Dist: locust>=2.31.5
+Requires-Dist: locust>=2.31.6
 Requires-Dist: oauthlib==3.2.2
 Requires-Dist: packaging==24.1
 Requires-Dist: platformdirs==4.2.2

{locust_cloud-1.0.7.dist-info → locust_cloud-1.0.9.dist-info}/RECORD

@@ -1,5 +1,5 @@
-locust_cloud/__init__.py,sha256=ZpA79bNf7scS9qM45j7YNHRYQqAd9MzrQKreYU0OmzE,1480
-locust_cloud/auth.py,sha256=5cIhUgaENeYSe_GxNp3v7rLhB4_Pj9URFrLYYsRG-n8,3042
+locust_cloud/__init__.py,sha256=A5I1PXZzB6FF2YgHq_hw4dLXUUt0FSkhAZG3lho1zng,1769
+locust_cloud/auth.py,sha256=kSSEbOcfZHNWmn802pAS01ooiNm1IIa29Ui_-rku4lM,3027
 locust_cloud/cloud.py,sha256=_TnyFaQG4rhd4msIs2I6QhAWOUDeRmcZXGzH4cEx2aI,12724
 locust_cloud/timescale/exporter.py,sha256=j1K-czMUdOLFt5Sop0PTF6TL52vYdngEEI5OEM3Xl60,11128
 locust_cloud/webui/.eslintrc,sha256=huuuTZNwbBGkhfelEOMhLQ-tgC9OYnnX0DRYgBn83Ig,1146
@@ -13,7 +13,7 @@ locust_cloud/webui/vite.config.ts,sha256=cqxPMkbwEA3H9mGGbuPulQUhIHCosUqm_1usxzs
 locust_cloud/webui/yarn.lock,sha256=yysZ-wROYB43fp1k_YC4zhLBjJKWr3_C6vKWM4rx0b4,216553
 locust_cloud/webui/dist/index.html,sha256=WiAeD-vj6jDZ1Q3-odnAPyM9hmc5pT6gvcq9YI65U9o,664
 locust_cloud/webui/dist/assets/index-CCS97q-k.js,sha256=3-KSz2kFRL7YG3rMB1dUCffEByaK7Zny5cuzxSgn5x8,2795908
-locust_cloud-1.0.7.dist-info/METADATA,sha256=zo39FGyY4bMlOwy7NW2aka7J-oo3H-tRhqDdqKxZ7aM,2162
-locust_cloud-1.0.7.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-locust_cloud-1.0.7.dist-info/entry_points.txt,sha256=PGyAb4e3aTsGS3N3VGShDl6VzJaXy7QwsEgsLOC7V00,57
-locust_cloud-1.0.7.dist-info/RECORD,,
+locust_cloud-1.0.9.dist-info/METADATA,sha256=KnYo_7ya1QDIh9sjeOfWM-6wzPOVLqHyi0Xt50J1Nl0,2162
+locust_cloud-1.0.9.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+locust_cloud-1.0.9.dist-info/entry_points.txt,sha256=PGyAb4e3aTsGS3N3VGShDl6VzJaXy7QwsEgsLOC7V00,57
+locust_cloud-1.0.9.dist-info/RECORD,,