Refactor cache value retrieval in Cache class to handle unpickling er…

…ror exception
jesse-ai · Mar 4, 2024 · dc9a9ca · dc9a9ca
1 parent cb2259c
commit dc9a9ca
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/jesse/services/cache.py b/jesse/services/cache.py
@@ -67,13 +67,13 @@ def get_value(self, key: str) -> Any:
             item['expire_at'] = time() + item['expire_seconds']
             self._update_db()
 
-        with open(item['path'], 'rb') as f:
-            try:
+        try:
+            with open(item['path'], 'rb') as f:
                 cache_value = pickle.load(f)
-            except EOFError:
-                # File got broken
-                cache_value = False
-            return cache_value
+        except (EOFError, pickle.UnpicklingError):
+            cache_value = False
+
+        return cache_value
 
     def _update_db(self) -> None:
         # store/update database