labere's memoizer

memoize.py

#!/usr/bin/env python

import var

""" this is labere's internal memoization. im only writing this myself because i didn't want to user someone elses, and im quite bored. """

def memoize(name):
    """ silly little decorator to get this done. 
        accepts a name to use for the dictionary to push/pull results from. """
    if not var.memoize.__contains__(name):
        var.memoize.update({name: {}})
    def wrap(f):
        """ wraps the function. """
        def memo(*args, **kw):
            """ catch the arguments, and finish the memoizing process.
                only except one arg. """
            if args[0] in var.memoize[name]:
                # print 'results cached' # debug
                return var.memoize[name][args[0]]
            else:
                # print 'caching results' # debug
                try: m = f(args[0], kw)
                except (TypeError): m = f(args[0])
                var.memoize[name][args[0]] = m
                return m
        return memo
    return wrap