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/utils/tracking/__init__.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

(limited to 'debug_toolbar/utils')

diff --git a/debug_toolbar/utils/tracking/__init__.py b/debug_toolbar/utils/tracking/__init__.py
index 4fbefa5..166b1b5 100644
--- a/debug_toolbar/utils/tracking/__init__.py
+++ b/debug_toolbar/utils/tracking/__init__.py
@@ -20,18 +20,21 @@ def pre_dispatch(func):
     return wrapped
 
 
-def replace_call(func):
+def replace_method(klass, method_name):
+    original = getattr(klass, method_name)
+
     def inner(callback):
         def wrapped(*args, **kwargs):
-            return callback(func, *args, **kwargs)
+            return callback(original, *args, **kwargs)
 
-        actual = getattr(func, '__wrapped__', func)
+        actual = getattr(original, '__wrapped__', original)
         wrapped.__wrapped__ = actual
         wrapped.__doc__ = getattr(actual, '__doc__', None)
         wrapped.__name__ = actual.__name__
 
-        _replace_function(func, wrapped)
+        setattr(klass, method_name, wrapped)
         return wrapped
+
     return inner
 
 
-- 
cgit v1.2.3