Removed redundant 'return', conditions in 'else/if' switched (to avoi…

…d using 'not'.
mate-academy · Nov 26, 2024 · c0004cc · c0004cc
1 parent d242dd1
commit c0004cc
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/app/main.py b/app/main.py
@@ -5,10 +5,11 @@ def cache(func: Callable) -> Callable:
     storage = {}
 
     def wrapper(*args) -> Callable:
-        if args not in storage:
-            print("Calculating new result")
-            return storage.setdefault(args, func(*args))
-        else:
+        if args in storage:
             print("Getting from cache")
-            return storage.get(args)
+        else:
+            print("Calculating new result")
+            storage.setdefault(args, func(*args))
+        return storage.get(args)
+
     return wrapper