From c8cd3e97408524ab9d76fcfcedd80ed1c4dfa840 Mon Sep 17 00:00:00 2001 From: Marcin Kuzminski Date: Wed, 4 Jul 2018 22:26:52 +0200 Subject: [PATCH] caches: use LRUDict inside our lru helper so we can import it later on when required. --- rhodecode/lib/memory_lru_debug.py | 21 ++++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/rhodecode/lib/memory_lru_debug.py b/rhodecode/lib/memory_lru_debug.py index cac647a2..81b4c952 100644 --- 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