fix

jxxghp · Jul 31, 2023 · c902b6a · c902b6a
1 parent 4d0b34f
commit c902b6a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/app/chain/__init__.py b/app/chain/__init__.py
@@ -38,15 +38,21 @@ def load_cache(filename: str) -> Any:
         """
         cache_path = settings.TEMP_PATH / filename
         if cache_path.exists():
-            return pickle.load(cache_path.open('rb'))
+            try:
+                return pickle.load(cache_path.open('rb'))
+            except Exception as err:
+                logger.error(f"加载缓存 {filename} 出错：{err}")
         return None
 
     @staticmethod
     def save_cache(cache: Any, filename: str) -> None:
         """
         保存缓存到本地
         """
-        pickle.dump(cache, (settings.TEMP_PATH / filename).open('wb'))
+        try:
+            pickle.dump(cache, (settings.TEMP_PATH / filename).open('wb'))
+        except Exception as err:
+            logger.error(f"保存缓存 {filename} 出错：{err}")
 
     def run_module(self, method: str, *args, **kwargs) -> Any:
         """