diff options
Diffstat (limited to 'debug_toolbar/panels/sql.py')
| -rw-r--r-- | debug_toolbar/panels/sql.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/debug_toolbar/panels/sql.py b/debug_toolbar/panels/sql.py index bc70b97..709a5f5 100644 --- a/debug_toolbar/panels/sql.py +++ b/debug_toolbar/panels/sql.py @@ -13,13 +13,12 @@ from debug_toolbar.panels import DebugPanel from debug_toolbar.utils import render_stacktrace from debug_toolbar.utils.sql import reformat_sql from debug_toolbar.utils.tracking.db import CursorWrapper -from debug_toolbar.utils.tracking import replace_call +from debug_toolbar.utils.tracking import replace_method -# Inject our tracking cursor -@replace_call(BaseDatabaseWrapper.cursor) -def cursor(func, self): - result = func(self) +@replace_method(BaseDatabaseWrapper, 'cursor') +def cursor(original, self): + result = original(self) djdt = DebugToolbarMiddleware.get_current() if not djdt: |