implement py-cache-decorator

mate-academy · Nov 12, 2024 · 6c6eaae · 6c6eaae
1 parent 5891c38
commit 6c6eaae
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/app/main.py b/app/main.py
@@ -2,5 +2,14 @@
 
 
 def cache(func: Callable) -> Callable:
-    # Write your code here
-    pass
+    cache_args = {}
+
+    def wrapper(*args, **kwargs) -> Callable:
+        key = (args, tuple(kwargs.items()))
+        if key in cache_args:
+            print("Getting from cache")
+        else:
+            print("Calculating new result")
+            cache_args[key] = func(*args, **kwargs)
+        return cache_args[key]
+    return wrapper