locust 2.31.7.dev2__py3-none-any.whl → 2.31.7.dev8__py3-none-any.whl

locust/_version.py

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

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:
@@ -686,7 +687,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

@@ -135,6 +135,11 @@ class WebUI:
         # ensures static js files work on Windows
         mimetypes.add_type("application/javascript", ".js")
 
+        if self.web_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()
         if not delayed_start:
@@ -516,13 +521,14 @@ class WebUI:
     @property
     def login_manager(self):
         if self.web_login:
-            login_manager = LoginManager()
-            login_manager.init_app(self.app)
-            login_manager.login_view = "login"
-            return login_manager
+            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)

{locust-2.31.7.dev2.dist-info → locust-2.31.7.dev8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.31.7.dev2
+Version: 2.31.7.dev8
 Summary: Developer-friendly load testing framework
 Home-page: https://locust.io/
 License: MIT

{locust-2.31.7.dev2.dist-info → locust-2.31.7.dev8.dist-info}/RECORD

@@ -1,10 +1,11 @@
 locust/__init__.py,sha256=Jit8eNUrwuMLqavyFvMZr69e61DILq_KB4yT4MciScw,1681
 locust/__main__.py,sha256=vBQ82334kX06ImDbFlPFgiBRiLIinwNk3z8Khs6hd74,31
-locust/_version.py,sha256=Q9T435Pf18gHeY2PAKDvCCK7vTh7paqLEkfA3cHQgJM,458
+locust/_version.py,sha256=5behDNp5CPxZh_tx28MZyxeIXF11-3XxmeoDl1-z9x4,458
 locust/argument_parser.py,sha256=V0EI7vVvAFLX68ntjRg6qGr2hXSLJgQ0y743zvurwA8,29008
 locust/clients.py,sha256=XK-xabq2_5GZKMEjebDobvEjeBTtCs8h2EelL7s68Qs,19346
 locust/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 locust/contrib/fasthttp.py,sha256=qKgUyaH1AppEj64LzcMu67tzS5yls8mELkKs-N5pEtQ,28431
+locust/contrib/mongodb.py,sha256=1seUYgJOaNKwybYOP9PUEVhgl8hGy-G33f8lFj3R8W8,1246
 locust/contrib/postgres.py,sha256=OuMWnGYN10K65Tq2axVESEW25Y0g5gJb0rK90jkcCJg,1230
 locust/debug.py,sha256=We6Z9W0btkKSc7PxWmrZx-xMynvOOsKhG6jmDgQin0g,5134
 locust/dispatch.py,sha256=prdwtb9EoN4A9klgiKgWuwQmvFB8hEuFHOK6ot62AJI,16202
@@ -14,7 +15,7 @@ locust/exception.py,sha256=jGgJ32ubuf4pWdlaVOkbh2Y0LlG0_DHi-lv3ib8ppOE,1791
 locust/html.py,sha256=ioHs8R6aqK_XCBSv4fOV90PYWv_MYQuW9Pl71R6EidM,3496
 locust/input_events.py,sha256=ZIyePyAMuA_YFYWg18g_pE4kwuQV3RbEB250MzXRwjY,3314
 locust/log.py,sha256=Wrkn0Ibugh5Sqjm4hGQ2-jUsy1tNMBdTctp4FyXQI24,3457
-locust/main.py,sha256=JpBo-TKv3O8mjh-6JPsF1jwLpJj9_UcRdBigGCK6OxU,29006
+locust/main.py,sha256=kWiQkFjDQ3h4ou40PQhAjma9WyOTnflO9VaO0QNgyF4,29002
 locust/py.typed,sha256=gkWLl8yD4mIZnNYYAIRM8g9VarLvWmTAFeUfEbxJLBw,65
 locust/rpc/__init__.py,sha256=nVGoHWFQxZjnhCDWjbgXIbmFbN9sizAjkhvSs9_642c,58
 locust/rpc/protocol.py,sha256=n-rb3GZQcAlldYDj4E4GuFGylYj_26GSS5U29meft5Y,1282
@@ -38,7 +39,7 @@ locust/util/load_locustfile.py,sha256=hn70KcIG8jHmZyuKv2pcEmwgWtOEu24Efeji1KRYNU
 locust/util/rounding.py,sha256=5haxR8mKhATqag6WvPby-MSRRgIw5Ob6thbyvMYZM7o,92
 locust/util/timespan.py,sha256=Y0LtnhUq2Mq19p04u0XtBlYQ_-S2cRvwRdgru8W9WhA,986
 locust/util/url.py,sha256=s_W2PCxvxTWxWX0yUvp-8VBuQm881KwI5X9iifogZG4,321
-locust/web.py,sha256=figneTKyRa95SUii3k4OQI_oQOfG3wQqQpOhOjKxxXw,27671
+locust/web.py,sha256=wwCIu2lCYGftuAwYZ2KXIXkkWGdgG0dDS5q3Tk6QEu0,27822
 locust/webui/dist/assets/favicon-dark.png,sha256=6zVkRtiRfU45qQGvEhf1cq2nNauFs_JW5SI79wT0YkM,2437
 locust/webui/dist/assets/favicon-light.png,sha256=VdG2GZyeTCOML7xfievupBP4EhmHoUqlVCw-tXqkvxU,2468
 locust/webui/dist/assets/index-DQd3Odi5.js,sha256=KSEsSxvrUSsT08C4AcXte5FdBqbQlsFeRfVCdw57h8g,1671822
@@ -46,8 +47,8 @@ locust/webui/dist/auth.html,sha256=lFK2hUASKiH4veqEKI8SVpVcv75L07iLgdsx1usfPRo,6
 locust/webui/dist/index.html,sha256=w1ar7WpzWHSVkgoL5n6IPCOvAgWfmOChoYofr2xsey0,654
 locust/webui/dist/report.html,sha256=bjpe_eyweJLqLk7REsQOdOPp7e8PoGRajD5P8gIJeNE,1472409
 poetry.lock,sha256=SYexv51dGkbiljj9e0x0LVzQ0-gxKn31hwhFTO2t298,205000
-locust-2.31.7.dev2.dist-info/LICENSE,sha256=78XGpIn3fHVBfaxlPNUfjVufSN7QsdhpJMRJHv2AFpo,1095
-locust-2.31.7.dev2.dist-info/METADATA,sha256=C92_BE2_cXGWCay9DKJEkvaJ2_e2AhAr7fmIT7gCMes,7724
-locust-2.31.7.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-locust-2.31.7.dev2.dist-info/entry_points.txt,sha256=RhIxlLwU_Ae_WjimS5COUDLagdCh6IOMyLtgaQxNmlM,43
-locust-2.31.7.dev2.dist-info/RECORD,,
+locust-2.31.7.dev8.dist-info/LICENSE,sha256=78XGpIn3fHVBfaxlPNUfjVufSN7QsdhpJMRJHv2AFpo,1095
+locust-2.31.7.dev8.dist-info/METADATA,sha256=e0BVX77F-xZXlJ7TwxTAydmZyZZfgOmZRzOETnnPb1g,7724
+locust-2.31.7.dev8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+locust-2.31.7.dev8.dist-info/entry_points.txt,sha256=RhIxlLwU_Ae_WjimS5COUDLagdCh6IOMyLtgaQxNmlM,43
+locust-2.31.7.dev8.dist-info/RECORD,,