Cache decorator solution

mate-academy · Sep 28, 2023 · d1c4c10 · d1c4c10
1 parent 5891c38
commit d1c4c10
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/app/main.py b/app/main.py
@@ -2,5 +2,13 @@
 
 
 def cache(func: Callable) -> Callable:
-    # Write your code here
-    pass
+    stored_runs = {}
+
+    def wrapper(*args):
+        if args in stored_runs:
+            print("Getting from cache")
+            return stored_runs[args]
+        print("Calculating new result")
+        stored_runs.update({args: func(*args)})
+        return stored_runs[args]
+    return wrapper