memoise.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	# -- 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