-
Notifications
You must be signed in to change notification settings - Fork 0
/
decorators.py
34 lines (28 loc) · 1002 Bytes
/
decorators.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
'''Taken from: wiki.python.org/moin/PythonDecoratorLibrary
'''
class Memoised(object):
'''Decorator that caches a function's return value each time it is called.
If called later with the same arguments, the cached value is returned, and
not re-evaluated.
See: http://en.wikipedia.org/wiki/Memoization#Automatic_memoization
'''
def __init__(self, func):
self.func = func
self.cache = {}
def __call__(self, *args):
try:
return self.cache[args]
except KeyError:
value = self.func(*args)
self.cache[args] = value
return value
except TypeError:
# uncachable -- for instance, passing a list as an argument.
# Better to not cache than to blow up entirely.
return self.func(*args)
def __repr__(self):
'''Return the function's docstring.
'''
return self.func.__doc__
# Alternate U.S. spelling.
memoized = memoised = Memoised