Solution

mate-academy · Nov 18, 2024 · 46ee03d · 46ee03d
1 parent 5891c38
commit 46ee03d
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/app/main.py b/app/main.py
@@ -1,6 +1,17 @@
-from typing import Callable
+from typing import Callable, Any
 
 
 def cache(func: Callable) -> Callable:
-    # Write your code here
-    pass
+    data = {}
+
+    def inner(*args) -> Any:
+        if args not in data:
+            print("Calculating new result")
+            result = func(*args)
+            data[args] = result
+            return result
+        if args in data:
+            print("Getting from cache")
+            return data[args]
+
+    return inner