From 1bf828463dfc48ea09409ae2c91176d5fb0d38c7 Mon Sep 17 00:00:00 2001
From: Aymeric Augustin
Date: Thu, 17 Oct 2013 09:42:19 +0200
Subject: Implement method replacement for Python 3.

Python 3 doesn't have unbound methods.
---
 debug_toolbar/panels/sql.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

(limited to 'debug_toolbar/panels')

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:
-- 
cgit v1.2.3