Solution cache decorator

mate-academy · Sep 27, 2023 · c7af899 · c7af899
1 parent 5891c38
commit c7af899
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/app/main.py b/app/main.py
@@ -1,6 +1,19 @@
-from typing import Callable
+from typing import Callable, Any
+import functools
 
 
 def cache(func: Callable) -> Callable:
-    # Write your code here
-    pass
+    memory = {}
+
+    @functools.wraps(func)
+    def wrapper_cache(*args) -> Any:
+        key = args
+        if key not in memory:
+            memory[key] = func(*args)
+            print("Calculating new result")
+            return memory[key]
+        else:
+            print("Getting from cache")
+            return memory[key]
+
+    return wrapper_cache