Improve cached decorator to support class based methods

Author: viniciuschiele

flask_apidoc/utils.py

@@ -2,22 +2,22 @@
 Helpers.
 """
 
+import functools
+
 
 def cached(f):
     """
     Cache decorator for functions taking one or more arguments.
     :param f: The function to be cached.
     :return: The cached value.
     """
-    class CachedDict(dict):
-        def __init__(self, f):
-            self.f = f
-
-        def __call__(self, *args):
-            return self[args]
 
-        def __missing__(self, key):
-            ret = self[key] = self.f(*key)
-            return ret
+    cache = f.cache = {}
 
-    return CachedDict(f)
+    @functools.wraps(f)
+    def decorator(*args, **kwargs):
+        key = str(args) + str(kwargs)
+        if key not in cache:
+            cache[key] = f(*args, **kwargs)
+        return cache[key]
+    return decorator