locust 2.31.4.dev27__py3-none-any.whl → 2.31.4.dev44__py3-none-any.whl

locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.31.4.dev27"
+__version__ = "2.31.4.dev44"
 version = __version__
-__version_tuple__ = (2, 31, 4, "dev27")
+__version_tuple__ = (2, 31, 4, "dev44")
 version_tuple = __version_tuple__

locust/contrib/postgres.py

@@ -0,0 +1,45 @@
+from locust import User, events
+
+import time
+
+import psycopg
+
+
+class PostgresClient:
+    def __init__(self, conn_string):
+        self.connection = psycopg.connect(conn_string)
+
+    def execute_query(self, query):
+        start_time = time.time()
+        try:
+            self.connection.execute(query)
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(
+                request_type="postgres_success",
+                name="execute_query",
+                response_time=response_time,
+                response_length=0,
+            )
+        except Exception as e:
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(
+                request_type="postgres_failure",
+                name="execute_query",
+                response_time=response_time,
+                response_length=0,
+                exception=e,
+            )
+
+    def close(self):
+        self.connection.close()
+
+
+class PostgresUser(User):
+    abstract = True
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.client = PostgresClient(conn_string=self.conn_string)
+
+    def on_stop(self):
+        self.client.close()

{locust-2.31.4.dev27.dist-info → locust-2.31.4.dev44.dist-info}/METADATA

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

{locust-2.31.4.dev27.dist-info → locust-2.31.4.dev44.dist-info}/RECORD

@@ -1,10 +1,11 @@
 locust/__init__.py,sha256=Jit8eNUrwuMLqavyFvMZr69e61DILq_KB4yT4MciScw,1681
 locust/__main__.py,sha256=vBQ82334kX06ImDbFlPFgiBRiLIinwNk3z8Khs6hd74,31
-locust/_version.py,sha256=vlPR-iGGBjEn1JWFNsWXzhNKIY5k4d442zYGhJ9qUk4,460
+locust/_version.py,sha256=0nomho3opGSVYhW7so7EWRipkaKCG-n_N4eTTXuqwPU,460
 locust/argument_parser.py,sha256=FMEWKC5OMVzpliRYa6KJvmpMRFvtlJFoVKxyW9_NJuc,29012
 locust/clients.py,sha256=OHPv6hBAt4gt3HI67yqyT1qrSsF8uMdCwIRu0kIsRWI,19491
 locust/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 locust/contrib/fasthttp.py,sha256=1hR2WK-3ACGVsU6Fu4Z4KucVzL1LXMLG6k54KeZdt78,28509
+locust/contrib/postgres.py,sha256=OuMWnGYN10K65Tq2axVESEW25Y0g5gJb0rK90jkcCJg,1230
 locust/debug.py,sha256=We6Z9W0btkKSc7PxWmrZx-xMynvOOsKhG6jmDgQin0g,5134
 locust/dispatch.py,sha256=esAp4k8j1zQ8pX2j01UiYUmWS1-9r-XhQvylYyyFeR0,16228
 locust/env.py,sha256=e5FNcpoftv8G0D86cR1t1cXatumBmCIDA7DbR1JjOSw,13000
@@ -44,8 +45,8 @@ locust/webui/dist/assets/index-BTVvoV6z.js,sha256=9_80rFccPA6Zj0O0Pltwe-ZlI0nxOl
 locust/webui/dist/auth.html,sha256=ZCBJy5NW6FPqa4V0Bq1lHzx26aSaF6UV21ffS0LH1Og,648
 locust/webui/dist/index.html,sha256=JVKEwn_J1HUJvBJa-9R1JNBpoXBimWMjfEoXdyXJogo,654
 locust/webui/dist/report.html,sha256=CPuv-WOa0QbAeJB57YapUdeb_hwoc65fe9M7N9GDLDs,1472463
-locust-2.31.4.dev27.dist-info/LICENSE,sha256=78XGpIn3fHVBfaxlPNUfjVufSN7QsdhpJMRJHv2AFpo,1095
-locust-2.31.4.dev27.dist-info/METADATA,sha256=O6_jCjsgMvrs3qAJNNPckB1SDdptkhPYT_oie3NwEac,7679
-locust-2.31.4.dev27.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
-locust-2.31.4.dev27.dist-info/entry_points.txt,sha256=RhIxlLwU_Ae_WjimS5COUDLagdCh6IOMyLtgaQxNmlM,43
-locust-2.31.4.dev27.dist-info/RECORD,,
+locust-2.31.4.dev44.dist-info/LICENSE,sha256=78XGpIn3fHVBfaxlPNUfjVufSN7QsdhpJMRJHv2AFpo,1095
+locust-2.31.4.dev44.dist-info/METADATA,sha256=faH7qAC7irfbTul1KToL5q6qF070Z0CJiWFK6VLwdTk,7679
+locust-2.31.4.dev44.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
+locust-2.31.4.dev44.dist-info/entry_points.txt,sha256=RhIxlLwU_Ae_WjimS5COUDLagdCh6IOMyLtgaQxNmlM,43
+locust-2.31.4.dev44.dist-info/RECORD,,