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

locust_cloud/__init__.py

@@ -12,6 +12,8 @@ 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):
@@ -38,8 +40,6 @@ 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/cloud.py

@@ -15,6 +15,7 @@ from botocore.exceptions import ClientError
 LAMBDA = "https://deployer.locust.cloud/1"
 DEFAULT_CLUSTER_NAME = "locust"
 DEFAULT_REGION_NAME = "eu-north-1"
+LOCUST_ENV_VARIABLE_IGNORE_LIST = ["LOCUST_BUILD_PATH", "LOCUST_SKIP_MONKEY_PATCH"]
 
 
 class LocustTomlConfigParser(configargparse.TomlConfigParser):
@@ -275,7 +276,7 @@ def deploy(
     locust_env_variables = [
         {"name": env_variable, "value": str(os.environ[env_variable])}
         for env_variable in os.environ
-        if env_variable.startswith("LOCUST_")
+        if env_variable.startswith("LOCUST_") and env_variable not in LOCUST_ENV_VARIABLE_IGNORE_LIST
     ]
 
     response = requests.post(

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}")

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

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

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

@@ -1,7 +1,7 @@
-locust_cloud/__init__.py,sha256=JNMaCykjuDy6HQoqokF-nr6gZegFXZy0Le12W8HILnQ,1484
+locust_cloud/__init__.py,sha256=ZpA79bNf7scS9qM45j7YNHRYQqAd9MzrQKreYU0OmzE,1480
 locust_cloud/auth.py,sha256=5cIhUgaENeYSe_GxNp3v7rLhB4_Pj9URFrLYYsRG-n8,3042
-locust_cloud/cloud.py,sha256=hsnLCjY_9WwgYu8uH_1p8zgPKUX-w7-ERhwuC6ErgEM,12584
-locust_cloud/timescale/exporter.py,sha256=jxLeRa_rAbzsO36mXKviOUsUyGCoS0dbd0PuvLsBUF4,11111
+locust_cloud/cloud.py,sha256=_TnyFaQG4rhd4msIs2I6QhAWOUDeRmcZXGzH4cEx2aI,12724
+locust_cloud/timescale/exporter.py,sha256=j1K-czMUdOLFt5Sop0PTF6TL52vYdngEEI5OEM3Xl60,11128
 locust_cloud/webui/.eslintrc,sha256=huuuTZNwbBGkhfelEOMhLQ-tgC9OYnnX0DRYgBn83Ig,1146
 locust_cloud/webui/.gitignore,sha256=i8EHIqDlVm1-Dkvf_GTZmP_Wu99GE7ABfbYzHXOU734,54
 locust_cloud/webui/.prettierrc,sha256=qZpzAJPlHUQ3--NHYK8WXgc_TlIqq-eoz4pbkTKK3nk,167
@@ -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.5.dist-info/METADATA,sha256=BLFEOcOxxPUE377hgX5Z7LVC6suYR7rwtJVlQaF8kEI,2162
-locust_cloud-1.0.5.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-locust_cloud-1.0.5.dist-info/entry_points.txt,sha256=PGyAb4e3aTsGS3N3VGShDl6VzJaXy7QwsEgsLOC7V00,57
-locust_cloud-1.0.5.dist-info/RECORD,,
+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,,