Décorateur de mémoisation de méthode de méthode Python Réinter

Je tente de construire un décorateur pour une méthode d'instance d'une classe qui tiendra le résultat. (Cela a été fait un million de fois auparavant) Cependant, j'aimerais pouvoir réinitialiser le cache mémo à tout moment (par exemple, si quelque chose dans l'état d'instance change, ce qui pourrait changer le résultat de la méthode n'ayant rien faire avec ses arguments). Ainsi, j'ai tenté de construire un décorateur comme une classe au lieu d'une fonction, de sorte que je pourrais avoir accès au cache comme membre de classe. Cela m'a conduit sur le chemin d'apprentissage sur les descripteurs, en particulier le __ obtenez __ code> méthode, ce qui est en fait bloqué. Mon code ressemble à:

import time

class memoized(object):

    def __init__(self, func):
        self.func = func
        self.cache = {}

    def __call__(self, *args, **kwargs):

        key = (self.func, args, frozenset(kwargs.iteritems()))

        try:
            return self.cache[key]
        except KeyError:
            self.cache[key] = self.func(*args, **kwargs)
            return self.cache[key]
        except TypeError:
            # uncacheable, so just return calculated value without caching
            return self.func(*args, **kwargs)

    # self == instance of memoized
    # obj == instance of my_class
    # objtype == class object of __main__.my_class
    def __get__(self, obj, objtype=None):
        """Support instance methods"""
        if obj is None:
            return self

        # new_func is the bound method my_func of my_class instance
        new_func = self.func.__get__(obj, objtype)

        # instantiates a brand new class...this is not helping us, because it's a 
        # new class each time, which starts with a fresh cache
        return self.__class__(new_func)

    # new method that will allow me to reset the memoized cache
    def reset(self):
        print "IN RESET"
        self.cache = {}

class my_class:
    @memoized
    def my_func(self, val):
        print "in my_func"
        time.sleep(2)
        return val


c = my_class()

print "should take time"
print c.my_func(55)
print

print "should be instant"
print c.my_func(55)
print

c.my_func.reset()

print "should take time"
print c.my_func(55)