Solution

app/main.py

@@ -1,6 +1,36 @@
-from typing import Callable
+from functools import wraps
+from typing import Callable, Any, List
 
 
 def cache(func: Callable) -> Callable:
-    # Write your code here
-    pass
+    stored_results = {}
+
+    @wraps(func)
+    def wrapper(*args, **kwargs) -> Any:
+        key = (args, tuple(sorted(kwargs.items())))
+        if key in stored_results:
+            print("Getting from cache")
+            return stored_results[key]
+        print("Calculating new result")
+        stored_results[key] = func(*args, **kwargs)
+        return stored_results[key]
+
+    return wrapper
+
+
+@cache
+def long_time_func(base: int, power: int, mod: int) -> int:
+    return (base ** power ** mod) % (base * mod)
+
+
+@cache
+def long_time_func_2(n_tuple: tuple, power: int) -> List[int]:
+    return [number ** power for number in n_tuple]
+
+
+long_time_func(1, 2, 3)
+long_time_func(2, 2, 3)
+long_time_func_2((5, 6, 7), 5)
+long_time_func(1, 2, 3)
+long_time_func_2((5, 6, 7), 10)
+long_time_func_2((5, 6, 7), 10)