plainx-sentry 0.5.0__py3-none-any.whl → 0.5.2__py3-none-any.whl

plainx/sentry/middleware.py

@@ -4,9 +4,9 @@ from sentry_sdk.tracing import TransactionSource
 from sentry_sdk.utils import capture_internal_exceptions
 
 try:
-    from plain.models.db import connection
+    from plain.models.db import connections, DEFAULT_DB_ALIAS
 except ImportError:
-    connection = None
+    connections = None
 
 import logging
 
@@ -19,11 +19,11 @@ def trace_db(execute, sql, params, many, context):
         data = {
             "db.params": params,
             "db.executemany": many,
-            "db.system": connection.vendor,
-            "db.name": connection.settings_dict.get("NAME"),
-            "db.user": connection.settings_dict.get("USER"),
-            "server.address": connection.settings_dict.get("HOST"),
-            "server.port": connection.settings_dict.get("PORT"),
+            "db.system": connections[DEFAULT_DB_ALIAS].vendor,
+            "db.name": connections[DEFAULT_DB_ALIAS].settings_dict.get("NAME"),
+            "db.user": connections[DEFAULT_DB_ALIAS].settings_dict.get("USER"),
+            "server.address": connections[DEFAULT_DB_ALIAS].settings_dict.get("HOST"),
+            "server.port": connections[DEFAULT_DB_ALIAS].settings_dict.get("PORT"),
         }
 
         sentry_sdk.add_breadcrumb(message=sql, category="query", data=data)
@@ -51,7 +51,7 @@ class SentryMiddleware:
             request_info = event.setdefault("request", {})
             request_info["url"] = request.build_absolute_uri()
             request_info["method"] = request.method
-            request_info["query_string"] = request.META.get("QUERY_STRING", "")
+            request_info["query_string"] = request.meta.get("QUERY_STRING", "")
             # Headers and env need some PII filtering, ideally,
             # among other filters... similar for GET/POST data?
             # request_info["headers"] = dict(request.headers)
@@ -81,9 +81,9 @@ class SentryMiddleware:
             with sentry_sdk.start_transaction(
                 op="http.server", name=request.path_info
             ) as transaction:
-                if connection:
+                if connections:
                     # Also get spans for db queries
-                    with connection.execute_wrapper(trace_db):
+                    with connections[DEFAULT_DB_ALIAS].execute_wrapper(trace_db):
                         response = self.get_response(request)
                 else:
                     # No db presumably
@@ -137,9 +137,9 @@ class SentryWorkerMiddleware:
                 name=f"job:{job.job_class}",
                 source=TransactionSource.TASK,
             ) as transaction:
-                if connection:
+                if connections:
                     # Also get spans for db queries
-                    with connection.execute_wrapper(trace_db):
+                    with connections[DEFAULT_DB_ALIAS].execute_wrapper(trace_db):
                         job_result = self.run_job(job)
                 else:
                     # No db presumably

{plainx_sentry-0.5.0.dist-info → plainx_sentry-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plainx-sentry
-Version: 0.5.0
+Version: 0.5.2
 Author-email: Dave Gaeddert <dave.gaeddert@gmail.com>
 Requires-Python: >=3.11
 Requires-Dist: sentry-sdk>=2.24.0

{plainx_sentry-0.5.0.dist-info → plainx_sentry-0.5.2.dist-info}/RECORD

@@ -1,9 +1,9 @@
 plainx/sentry/__init__.py,sha256=Klz3mH_itM2RXzJ_sWWJRIoZcS0jCRxWrhVVIPsG5FY,123
 plainx/sentry/config.py,sha256=EwMnY3DaanIGkgtKD_4h8ncx6MH_MxOML8BF5Hx5GBA,723
 plainx/sentry/default_settings.py,sha256=DNX6OQ0-BqfjyWuszW7JnZhiDn0tGtS3Fa2W7VbTIF8,564
-plainx/sentry/middleware.py,sha256=yA5gWAcyfP30VitaXrUJArfHKC4O3_hr9uN0HMZA4Mo,6019
+plainx/sentry/middleware.py,sha256=huo7bEmz-yP1ft8vvMHfJqUN5G3tZlZfwLbkBwjnlQI,6174
 plainx/sentry/templates.py,sha256=YovKIswcsjLRjHdmRzdaxG1PHjGfz8fc5TeNE9wcYQA,2046
 plainx/sentry/templates/sentry/js.html,sha256=aOjWAwuUaecGbC-5yZWP1aaGyYAcM-GK0dru-4VOYoE,491
-plainx_sentry-0.5.0.dist-info/METADATA,sha256=mv4iGcs5UH9Ka_MjVnP9G_Zxe2tjsQZDL_uewAaokQ8,1559
-plainx_sentry-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-plainx_sentry-0.5.0.dist-info/RECORD,,
+plainx_sentry-0.5.2.dist-info/METADATA,sha256=h3I8SOjZWC6rfW_Jhyeax594li1llkyhTpYQQMDfgSg,1559
+plainx_sentry-0.5.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+plainx_sentry-0.5.2.dist-info/RECORD,,