locust 2.31.6.dev23__py3-none-any.whl → 2.31.7__py3-none-any.whl

locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.31.6.dev23"
+__version__ = "2.31.7"
 version = __version__
-__version_tuple__ = (2, 31, 6, "dev23")
+__version_tuple__ = (2, 31, 7)
 version_tuple = __version_tuple__

locust/argument_parser.py

@@ -424,6 +424,12 @@ def setup_parser_arguments(parser):
         help="Host to bind the web interface to. Defaults to '*' (all interfaces)",
         env_var="LOCUST_WEB_HOST",
     )
+    web_ui_group.add_argument(
+        "--web-host-display-name",
+        type=str,
+        help=configargparse.SUPPRESS,
+        env_var="LOCUST_WEB_HOST_DISPLAY_NAME",
+    )
     web_ui_group.add_argument(
         "--web-port",
         "-P",

locust/contrib/mongodb.py

@@ -0,0 +1,41 @@
+from locust import User, events
+
+import time
+
+from pymongo import MongoClient
+from pymongo.errors import PyMongoError
+
+
+class MongoDBClient(MongoClient):
+    def __init__(self, conn_string, db_name):
+        super().__init__(conn_string)
+        self.db = self.client[db_name]
+
+    def execute_query(self, collection_name, query):
+        start_time = time.time()
+        try:
+            collection = self.db[collection_name]
+            collection.find(query)
+
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(request_type="MONGODB", name="QUERY", response_time=response_time, response_length=0)
+        except PyMongoError as e:
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(
+                request_type="MONGODB",
+                name="QUERY",
+                response_time=response_time,
+                response_length=0,
+                exception=e,
+            )
+
+
+class MongoDBUser(User):
+    abstract = True
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.client = MongoDBClient(conn_string=self.conn_string, db_name=self.db_name)
+
+    def on_stop(self):
+        self.client.close()

locust/main.py

@@ -199,6 +199,8 @@ def main():
             sys.exit(1)
 
     children = []
+    logger = logging.getLogger(__name__)
+    logger.info(f"Starting Locust {version}")
 
     if options.processes:
         if os.name == "nt":
@@ -298,7 +300,6 @@ def main():
 
                 atexit.register(kill_workers, children)
 
-    logger = logging.getLogger(__name__)
     greenlet_exception_handler = greenlet_exception_logger(logger)
 
     if options.stop_timeout:
@@ -486,6 +487,8 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
             web_host = options.web_host
         if web_host:
             logger.info(f"Starting web interface at {protocol}://{web_host}:{options.web_port}")
+        if options.web_host_display_name:
+            logger.info(f"Starting web interface at {options.web_host_display_name}")
         else:
             if os.name == "nt":
                 logger.info(
@@ -686,7 +689,6 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
     gevent.signal_handler(signal.SIGTERM, sig_term_handler)
 
     try:
-        logger.info(f"Starting Locust {version}")
         if options.class_picker:
             logger.info("Locust is running with the UserClass Picker Enabled")
         if options.autostart and not options.headless:

locust/web.py

@@ -136,9 +136,9 @@ class WebUI:
         mimetypes.add_type("application/javascript", ".js")
 
         if self.web_login:
-            self.login_manager = LoginManager()
-            self.login_manager.init_app(app)
-            self.login_manager.login_view = "login"
+            self._login_manager = LoginManager()
+            self._login_manager.init_app(self.app)
+            self._login_manager.login_view = "login"
 
         if environment.runner:
             self.update_template_args()
@@ -150,7 +150,7 @@ class WebUI:
             error_message = str(error)
             error_code = getattr(error, "code", 500)
             logger.log(
-                logging.INFO if error_code <= 404 else logging.ERROR,
+                logging.DEBUG if error_code <= 404 else logging.ERROR,
                 f"UI got request for {request.path}, but it resulted in a {error_code}: {error.name}",
             )
             return make_response(error_message, error_code)
@@ -518,6 +518,17 @@ class WebUI:
 
             return {}, 201
 
+    @property
+    def login_manager(self):
+        if self.web_login:
+            return self._login_manager
+
+        raise AttributeError("The login_manager is only available with --web-login.\n")
+
+    @login_manager.setter
+    def login_manager(self, value):
+        self._login_manager = value
+
     def start(self):
         self.greenlet = gevent.spawn(self.start_server)
         self.greenlet.link_exception(greenlet_exception_handler)