locust-cloud 1.0.4__py3-none-any.whl → 1.0.6__py3-none-any.whl

locust_cloud/__init__.py

@@ -6,8 +6,6 @@ import os
 from locust import events
 from locust.argument_parser import LocustArgumentParser
 
-os.environ["LOCUST_BUILD_PATH"] = os.path.join(os.path.dirname(__file__), "webui/dist")
-os.environ["LOCUST_WEB_LOGIN"] = "1"
 PG_USER = os.environ.get("PG_USER")
 PG_HOST = os.environ.get("PG_HOST")
 PG_PASSWORD = os.environ.get("PG_PASSWORD")
@@ -40,6 +38,8 @@ def add_arguments(parser: LocustArgumentParser):
 
 @events.init.add_listener
 def on_locust_init(environment, **args):
+    os.environ["LOCUST_BUILD_PATH"] = os.path.join(os.path.dirname(__file__), "webui/dist")
+
     if environment.parsed_options.exporter:
         Timescale(
             environment,

locust_cloud/timescale/exporter.py

@@ -25,7 +25,7 @@ def safe_serialize(obj):
 class Timescale:
     def __init__(self, environment: locust.env.Environment, pg_user, pg_host, pg_password, pg_database, pg_port):
         self.env = environment
-        self._run_id = ""
+        self._run_id = None
         self._samples: list[dict] = []
         self._background = gevent.spawn(self._run)
         self._hostname = socket.gethostname()
@@ -75,7 +75,7 @@ class Timescale:
 
     def on_test_start(self, environment: locust.env.Environment):
         if not self.env.parsed_options or not self.env.parsed_options.worker:
-            environment._run_id = datetime.now(UTC)  # type: ignore
+            self._run_id = environment._run_id = datetime.now(UTC)  # type: ignore
             msg = environment._run_id.strftime("%Y-%m-%d, %H:%M:%S.%f")  # type: ignore
             if environment.runner is not None:
                 logging.debug(f"about to send run_id to workers: {msg}")