Implement method replacement for Python 3.

Python 3 doesn't have unbound methods.

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: