Added a database stats tracking class to pass through params. This is setup

for a later addition of being able to add EXPLAIN support to individual queries
for SQL debugging.  Contributed by David Cramer.

1 files changed, 24 insertions, 1 deletions

diff --git a/debug_toolbar/panels/sql.py b/debug_toolbar/panels/sql.py
index d699e1a..2ca50ed 100644
--- a/debug_toolbar/panels/sql.py
+++ b/debug_toolbar/panels/sql.py
@@ -1,7 +1,29 @@
+from time import time
 from debug_toolbar.panels import DebugPanel
 from django.db import connection
+from django.db.backends import util
 from django.template.loader import render_to_string
 
+class DatabaseStatTracker(util.CursorDebugWrapper):
+    """
+    Replacement for CursorDebugWrapper which stores additional information
+    in `connection.queries`.
+    """
+    def execute(self, sql, params=()):
+        start = time()
+        try:
+            return self.cursor.execute(sql, params)
+        finally:
+            stop = time()
+            # We keep `sql` to maintain backwards compatibility
+            self.db.queries.append({
+                'sql': self.db.ops.last_executed_query(self.cursor, sql, params),
+                'time': stop - start,
+                'raw_sql': sql,
+                'params': params,
+            })
+util.CursorDebugWrapper = DatabaseStatTracker
+
 class SQLDebugPanel(DebugPanel):
     """
     Panel that displays information about the SQL queries run while processing the request.
@@ -9,7 +31,8 @@ class SQLDebugPanel(DebugPanel):
     name = 'SQL'
     
     def title(self):
-        return '%d SQL Queries' % (len(connection.queries))
+        total_time = sum(map(lambda q: float(q['time']) * 1000, connection.queries))
+        return '%d SQL Queries (%.2fms)' % (len(connection.queries), total_time)
 
     def url(self):
         return ''