Refactored lrudecorator using FunctionCacheManager.

1 files changed, 2 insertions, 14 deletions

diff --git a/pylru.py b/pylru.py
index 1e23ba2..5cb4130 100644
--- a/pylru.py
+++ b/pylru.py
@@ -534,19 +534,7 @@ def lruwrap(store, size, writeback=False):
 
 class lrudecorator(object):
     def __init__(self, size):
-        self.cache = lrucache(size)
+        self.size = size
 
     def __call__(self, func):
-        def wrapped(*args, **kwargs):
-            kwtuple = tuple((key, kwargs[key]) for key in sorted(kwargs.keys()))
-            key = (args, kwtuple)
-            try:
-                return self.cache[key]
-            except KeyError:
-                pass
-
-            value = func(*args, **kwargs)
-            self.cache[key] = value
-            return value
-        wrapped.cache = self.cache
-        return wrapped
+        return FunctionCacheManager(func, self.size)