locust 2.32.3.dev6__py3-none-any.whl → 2.32.3.dev11__py3-none-any.whl

locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.32.3.dev6"
+__version__ = "2.32.3.dev11"
 version = __version__
-__version_tuple__ = (2, 32, 3, "dev6")
+__version_tuple__ = (2, 32, 3, "dev11")
 version_tuple = __version_tuple__

locust/argument_parser.py

@@ -59,15 +59,18 @@ class LocustArgumentParser(configargparse.ArgumentParser):
         Arguments:
             include_in_web_ui: If True (default), the argument will show in the UI.
             is_secret: If True (default is False) and include_in_web_ui is True, the argument will show in the UI with a password masked text input.
+            is_required: If True (default is False) and include_in_web_ui is True, the argument will show in the UI as a required form field.
 
         Returns:
             argparse.Action: the new argparse action
         """
         include_in_web_ui = kwargs.pop("include_in_web_ui", True)
         is_secret = kwargs.pop("is_secret", False)
+        is_required = kwargs.pop("is_required", False)
         action = super().add_argument(*args, **kwargs)
         action.include_in_web_ui = include_in_web_ui
         action.is_secret = is_secret
+        action.is_required = is_required
         return action
 
     @property
@@ -82,6 +85,14 @@ class LocustArgumentParser(configargparse.ArgumentParser):
             if a.dest in self.args_included_in_web_ui and hasattr(a, "is_secret") and a.is_secret
         }
 
+    @property
+    def required_args_included_in_web_ui(self) -> dict[str, configargparse.Action]:
+        return {
+            a.dest: a
+            for a in self._actions
+            if a.dest in self.args_included_in_web_ui and hasattr(a, "is_required") and a.is_required
+        }
+
 
 class LocustTomlConfigParser(configargparse.TomlConfigParser):
     def parse(self, stream):
@@ -798,6 +809,7 @@ def default_args_dict() -> dict:
 class UIExtraArgOptions(NamedTuple):
     default_value: str
     is_secret: bool
+    is_required: bool
     help_text: str
     choices: list[str] | None = None
 
@@ -813,6 +825,7 @@ def ui_extra_args_dict(args=None) -> dict[str, dict[str, Any]]:
         k: UIExtraArgOptions(
             default_value=v,
             is_secret=k in parser.secret_args_included_in_web_ui,
+            is_required=k in parser.required_args_included_in_web_ui,
             help_text=parser.args_included_in_web_ui[k].help,
             choices=parser.args_included_in_web_ui[k].choices,
         )._asdict()

locust/log.py

@@ -92,7 +92,7 @@ def greenlet_exception_logger(logger, level=logging.CRITICAL):
     """
 
     def exception_handler(greenlet):
-        if greenlet.exc_info[0] == SystemExit:
+        if greenlet.exc_info[0] is SystemExit:
             logger.log(
                 min(logging.INFO, level),  # dont use higher than INFO for this, because it sounds way to urgent
                 f"sys.exit({greenlet.exc_info[1]}) called (use log level DEBUG for callstack)",

locust/runners.py

@@ -248,7 +248,7 @@ class Runner:
                         "While stopping users, we encountered a user that didn't have proper args %s", user_greenlet
                     )
                     continue
-                if type(user) == self.user_classes_by_name[user_class]:
+                if type(user) is self.user_classes_by_name[user_class]:
                     to_stop.append(user)
 
             if not to_stop:

locust/web.py

@@ -60,6 +60,7 @@ class InputField(TypedDict, total=False):
     default_value: bool | None
     choices: list[str] | None
     is_secret: bool | None
+    is_required: bool | None
 
 
 class CustomForm(TypedDict, total=False):