memoise.py
fb1efef397b2iugw2017

memoise.py
View Options

			# -- coding: utf-8 --
			##
			## This file is part of Invenio.
			## Copyright (C) 2013 CERN.
			##
			## Invenio is free software; you can redistribute it and/or
			## modify it under the terms of the GNU General Public License as
			## published by the Free Software Foundation; either version 2 of the
			## License, or (at your option) any later version.
			##
			## Invenio is distributed in the hope that it will be useful, but
			## WITHOUT ANY WARRANTY; without even the implied warranty of
			## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
			## General Public License for more details.
			##
			## You should have received a copy of the GNU General Public License
			## along with Invenio; if not, write to the Free Software Foundation, Inc.,
			## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.

			"""
			Memoisation utilities.
			"""
			import functools


			class Memoise:
			"""
			Basic memoisation helper.
			Usage: fun = Memoise(fun)
			"""

			def __init__(self, function):
			"""Initialise."""
			self.memo = {}
			self.function = function

			def __call__(self, *args):
			"""Run and eventually memoise."""
			if args not in self.memo:
			self.memo[args] = self.function(*args)
			return self.memo[args]


			def memoize(obj):
			cache = obj.cache = {}
			@functools.wraps(obj)
			def memoizer(args, *kwargs):
			key = str(args) + str(kwargs)
			if key not in cache:
			cache[key] = obj(args, *kwargs)
			return cache[key]
			return memoizer