Subclass OperationDebugWrapper to log to info

Author: aclark4life

django_mongodb_extensions/debug_toolbar/panels/mql/tracking.py

@@ -7,6 +7,8 @@
 
 import django.test.testcases
 from django.utils.encoding import force_str
+from django.conf import settings
+from django.db.backends.utils import logger
 
 from debug_toolbar.utils import get_stack_trace, get_template_info
 
@@ -19,6 +21,32 @@
 allow_sql = contextvars.ContextVar("debug-toolbar-allow-sql", default=True)
 
 
+
+class DebugToolbarWrapper(OperationDebugWrapper):
+    def log(self, op, duration, args, kwargs=None):
+        msg = "(%.3f) %s"
+        args = ", ".join(repr(arg) for arg in args)
+        operation = f"db.{self.collection_name}{op}({args})"
+        if len(settings.DATABASES) > 1:
+            msg += f"; alias={self.db.alias}"
+        self.db.queries_log.append(
+            {
+                "sql": operation,
+                "time": "%.3f" % duration,
+            }
+        )
+        logger.info(
+            msg,
+            duration,
+            operation,
+            extra={
+                "duration": duration,
+                "sql": operation,
+                "alias": self.db.alias,
+            },
+        )
+
+
 class SQLQueryTriggered(Exception):
     """Thrown when template panel triggers a query"""
 
@@ -27,7 +55,7 @@ def patch_get_collection(connection):
 
     def get_collection(self, name, **kwargs):
         collection = Collection(self.database, name, **kwargs)
-        collection = OperationDebugWrapper(self, collection)
+        collection = DebugToolbarWrapper(self, collection)
         return collection
 
     if not hasattr(connection, "_djdt_cursor"):