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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
locust_cloud/__init__.py CHANGED
@@ -12,6 +12,8 @@ PG_PASSWORD = os.environ.get("PG_PASSWORD")
12
12
  PG_DATABASE = os.environ.get("PG_DATABASE")
13
13
  PG_PORT = os.environ.get("PG_PORT", 5432)
14
14
 
15
+ os.environ["LOCUST_BUILD_PATH"] = os.path.join(os.path.dirname(__file__), "webui/dist")
16
+
15
17
 
16
18
  @events.init_command_line_parser.add_listener
17
19
  def add_arguments(parser: LocustArgumentParser):
@@ -38,8 +40,6 @@ def add_arguments(parser: LocustArgumentParser):
38
40
 
39
41
  @events.init.add_listener
40
42
  def on_locust_init(environment, **args):
41
- os.environ["LOCUST_BUILD_PATH"] = os.path.join(os.path.dirname(__file__), "webui/dist")
42
-
43
43
  if environment.parsed_options.exporter:
44
44
  Timescale(
45
45
  environment,
locust_cloud/cloud.py CHANGED
@@ -15,6 +15,7 @@ from botocore.exceptions import ClientError
15
15
  LAMBDA = "https://deployer.locust.cloud/1"
16
16
  DEFAULT_CLUSTER_NAME = "locust"
17
17
  DEFAULT_REGION_NAME = "eu-north-1"
18
+ LOCUST_ENV_VARIABLE_IGNORE_LIST = ["LOCUST_BUILD_PATH", "LOCUST_SKIP_MONKEY_PATCH"]
18
19
 
19
20
 
20
21
  class LocustTomlConfigParser(configargparse.TomlConfigParser):
@@ -275,7 +276,7 @@ def deploy(
275
276
  locust_env_variables = [
276
277
  {"name": env_variable, "value": str(os.environ[env_variable])}
277
278
  for env_variable in os.environ
278
- if env_variable.startswith("LOCUST_")
279
+ if env_variable.startswith("LOCUST_") and env_variable not in LOCUST_ENV_VARIABLE_IGNORE_LIST
279
280
  ]
280
281
 
281
282
  response = requests.post(