diff --git a/rhodecode/lib/memory_lru_debug.py b/rhodecode/lib/memory_lru_debug.py --- a/rhodecode/lib/memory_lru_debug.py +++ b/rhodecode/lib/memory_lru_debug.py @@ -36,13 +36,32 @@ eg difference in speed: """ import logging -from infrae.cache.beakerext.lru import LRUDict +from repoze.lru import LRUCache from beaker.container import MemoryNamespaceManager, AbstractDictionaryNSManager from rhodecode.lib.utils2 import safe_str log = logging.getLogger(__name__) +class LRUDict(LRUCache): + """ Wrapper to provide partial dict access + """ + def __setitem__(self, key, value): + return self.put(key, value) + + def __getitem__(self, key): + return self.get(key) + + def __contains__(self, key): + return bool(self.get(key)) + + def __delitem__(self, key): + del self.data[key] + + def keys(self): + return self.data.keys() + + class LRUDictDebug(LRUDict): """ Wrapper to provide some debug options