Skip to content

Instantly share code, notes, and snippets.

@zzl0
Created March 31, 2014 05:44
Show Gist options
  • Save zzl0/9885985 to your computer and use it in GitHub Desktop.
Save zzl0/9885985 to your computer and use it in GitHub Desktop.
class memoized(object):
'''Decorator. Caches a function's return value each time it is called.
If called later with the same arguments, the cached value is returned
(not reevaluated).
'''
def __init__(self, func):
self.func = func
self.cache = {}
def __call__(self, *args):
if not isinstance(args, collections.Hashable):
# uncacheable. a list, for instance.
# better to not cache than blow up.
return self.func(*args)
if args in self.cache:
return self.cache[args]
else:
value = self.func(*args)
self.cache[args] = value
return value
def __repr__(self):
'''Return the function's docstring.'''
return self.func.__doc__
def __get__(self, obj, objtype):
'''Support instance methods.'''
return functools.partial(self.__call__, obj)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment