diff options
| -rw-r--r-- | debug_toolbar/panels/sql/tracking.py | 9 |
1 files changed, 6 insertions, 3 deletions
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) |