memorize_instance_method.py

memorize_instance_method.py

import random
import functools
import json
import weakref
import time
import sys
import gevent

class memoize(object):
    def __init__(self, function):
        self._caches = weakref.WeakKeyDictionary()
        self._function = function

        @functools.wraps(function)
        def wrapd(instance, *args, **kw):
            if instance not in self._caches:
                self._caches[instance] = {}
            cache_key = json.dumps([args, kw], sort_keys=True)
            if cache_key not in self._caches[instance]:
                self._caches[instance][cache_key] = self._function(instance, *args, **kw)
            return self._caches[instance][cache_key]
        self.wrapd = wrapd

    def __get__(self, instance, cls=None):
        if instance:
            return functools.partial(self.wrapd, instance)
        return self