podhmo · August 4, 2012 06:57 · Aug 4, 2012 · Aug 4, 2012
diff --git a/intercept.py b/intercept.py
@@ -16,11 +16,8 @@ def __init__(self, name):
 
 def intercept(cls, method):
     def doaction(caller, *args, **kwargs):
-        original_type = caller.__class__
         caller.__class__ = cls
-        v = method(caller, *args, **kwargs)
-        caller.__class = original_type
-        return v
+        return method.im_func(caller, *args, **kwargs)
     return doaction
 
 

diff --git a/intercept.py b/intercept.py
@@ -0,0 +1,38 @@
+class A(object):
+    fmt = "tweets %s"
+    def hello(self):
+        return "hello " + self.name
+
+    @classmethod
+    def tweets(cls, mes):
+        return cls.fmt % mes
+    def __init__(self, name):
+        self.name = name
+
+class B(object):
+    fmt = "called with B: %s"
+    def __init__(self, name):
+        self.name = name
+
+def intercept(cls, method):
+    def doaction(caller, *args, **kwargs):
+        original_type = caller.__class__
+        caller.__class__ = cls
+        v = method(caller, *args, **kwargs)
+        caller.__class = original_type
+        return v
+    return doaction
+
+
+print A("foo").hello()
+try:
+    print A.hello(B("bar"))
+except TypeError, e:
+    print e
+print intercept(A, A.hello)(B("bar"))
+
+print "=="
+
+print A.tweets("hey")
+B.tweets = A.__dict__["tweets"]
+print B.tweets("bar")