From 8ef3b8cdfb6ce69c0eccb030124b0b197b202ba8 Mon Sep 17 00:00:00 2001
From: Aymeric Augustin
Date: Fri, 29 Nov 2013 22:34:13 +0100
Subject: Record SQL from executemany.

Fix #293.
---
 debug_toolbar/panels/sql/tracking.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'debug_toolbar')

diff --git a/debug_toolbar/panels/sql/tracking.py b/debug_toolbar/panels/sql/tracking.py
index d412872..2ae65c7 100644
--- a/debug_toolbar/panels/sql/tracking.py
+++ b/debug_toolbar/panels/sql/tracking.py
@@ -98,10 +98,10 @@ class NormalCursorWrapper(object):
         except UnicodeDecodeError:
             return '(encoded string)'
 
-    def execute(self, sql, params=()):
+    def _record(self, method, sql, params):
         start_time = time()
         try:
-            return self.cursor.execute(sql, params)
+            return method(sql, params)
         finally:
             stop_time = time()
             duration = (stop_time - start_time) * 1000
@@ -171,8 +171,11 @@ class NormalCursorWrapper(object):
             # We keep `sql` to maintain backwards compatibility
             self.logger.record(**params)
 
+    def execute(self, sql, params=()):
+        return self._record(self.cursor.execute, sql, params)
+
     def executemany(self, sql, param_list):
-        return self.cursor.executemany(sql, param_list)
+        return self._record(self.cursor.executemany, sql, param_list)
 
     def __getattr__(self, attr):
         return getattr(self.cursor, attr)
-- 
cgit v1.2.3