Decorate methods per instance in Python

Question

Suppose I have a simple class

class TestClass:
    def doSomething(self):
        print 'Did something'

I would like to decorate a method doSomething, for example, to count the number of calls

class SimpleDecorator(object):
    def __init__(self,func):
        self.func=func
        self.count=0
    def __get__(self,obj,objtype=None):
        return MethodType(self,obj,objtype)
    def __call__(self,*args,**kwargs):
        self.count+=1
        return self.func(*args,**kwargs)

Now this counts the number of calls to the decorated method, however I would like to have a counter for each instance, so after

foo1=TestClass()
foo1.doSomething()
foo2=TestClass()

foo1.doSomething.countequals 1 and foo2.doSomething.countequals 0. From what I understand, this is not possible with the help of decorators. Is there any way to achieve this behavior?

Answer 1

Use the fact that self(i.e. the object to which the method is called) is passed as a parameter to the method:

import functools

def counted(method):
    @functools.wraps(method)
    def wrapped(obj, *args, **kwargs):
        if hasattr(obj, 'count'): 
            obj.count += 1
        else:
            obj.count = 1
        return method(obj, *args, **kwargs)
    return wrapped

obj . :

class Foo(object):
    @counted
    def do_something(self): pass

Answer 2

*args , ? ?