locust 2.42.6.dev3__tar.gz → 2.42.6.dev7__tar.gz

{locust-2.42.6.dev3 → locust-2.42.6.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.42.6.dev3
+Version: 2.42.6.dev7
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust

{locust-2.42.6.dev3 → locust-2.42.6.dev7}/locust/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.42.6.dev3'
-__version_tuple__ = version_tuple = (2, 42, 6, 'dev3')
+__version__ = version = '2.42.6.dev7'
+__version_tuple__ = version_tuple = (2, 42, 6, 'dev7')
 
 __commit_id__ = commit_id = None

{locust-2.42.6.dev3 → locust-2.42.6.dev7}/locust/argument_parser.py

@@ -151,6 +151,14 @@ class LocustTomlConfigParser(configargparse.TomlConfigParser):
         return result
 
 
+class LocustConfigParser(configargparse.ConfigFileParser):
+    def parse(self, stream):
+        if stream.name.endswith(".toml"):
+            return LocustTomlConfigParser(["tool.locust"]).parse(stream)
+
+        return configargparse.DefaultConfigFileParser().parse(stream)
+
+
 def parse_locustfile_paths(paths: list[str]) -> list[str]:
     """
     Returns a list of relative file paths.
@@ -218,12 +226,7 @@ def download_locustfile_from_url(url: str) -> str:
 def get_empty_argument_parser(add_help=True, default_config_files=DEFAULT_CONFIG_FILES) -> LocustArgumentParser:
     parser = LocustArgumentParser(
         default_config_files=default_config_files,
-        config_file_parser_class=configargparse.CompositeConfigParser(
-            [
-                LocustTomlConfigParser(["tool.locust"]),
-                configargparse.DefaultConfigFileParser,
-            ]
-        ),
+        config_file_parser_class=LocustConfigParser,
         add_env_var_help=False,
         add_config_file_help=False,
         add_help=add_help,