sol

app/main.py

@@ -1,6 +1,26 @@
-from typing import Callable
+from typing import Callable, Any
+from functools import wraps
 
 
 def cache(func: Callable) -> Callable:
-    # Write your code here
-    pass
+    cached = []
+    arg_cache = []
+
+    @wraps(func)
+    def wrapper(*args) -> Any:
+        nonlocal cached, arg_cache
+
+        if args not in arg_cache:
+            print("Calculating new result")
+            result = func(*args)
+            cached.append(result)
+            arg_cache.append(args)
+
+            return result
+
+        else:
+            print("Getting from cache")
+
+        return cached[arg_cache.index(args)]
+
+    return wrapper