caching decorator

memoize.py

def memoize(f):
    cache = {}

    def memoized_function(*args):
        if args not in cache:
            cache[args] = f(*args)
        return cache[args]

    memoized_function.cache = cache
    return memoized_function


@memoize
def foo(a, b):
    return a + b