Adjusting the exception handling mechanism

Author: peng3307165

modelcache/adapter/adapter.py

@@ -21,8 +21,7 @@ def cache_data_convert(cache_data, cache_query):
                 **kwargs
             )
         except Exception as e:
-            return 'adapt_query_exception'
-
+            return str(e)
 
     @classmethod
     def create_insert(cls, *args, **kwargs):
@@ -32,9 +31,7 @@ def create_insert(cls, *args, **kwargs):
                 **kwargs
             )
         except Exception as e:
-            logging.info('adapt_insert_e: {}'.format(e))
-            return 'adapt_insert_exception'
-
+            return str(e)
 
     @classmethod
     def create_remove(cls, *args, **kwargs):
@@ -45,7 +42,7 @@ def create_remove(cls, *args, **kwargs):
             )
         except Exception as e:
             logging.info('adapt_remove_e: {}'.format(e))
-            return 'adapt_remove_exception'
+            return str(e)
 
 
 def construct_resp_from_cache(return_message, return_query):

modelcache/adapter/adapter_remove.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 from modelcache import cache
-from modelcache.utils.error import NotInitError
+from modelcache.utils.error import NotInitError, RemoveError
 
 
 def adapt_remove(*args, **kwargs):
@@ -17,11 +17,10 @@ def adapt_remove(*args, **kwargs):
     if remove_type == 'delete_by_id':
         id_list = kwargs.pop("id_list", [])
         resp = chat_cache.data_manager.delete(id_list, model=model)
-
     elif remove_type == 'truncate_by_model':
         resp = chat_cache.data_manager.truncate(model)
-
     else:
-        resp = "remove_type_error"
+        # resp = "remove_type_error"
+        raise RemoveError()
     return resp
 

modelcache/utils/error.py

@@ -9,6 +9,11 @@ def __init__(self):
         super().__init__("The cache should be inited before using")
 
 
+class RemoveError(CacheError):
+    """Raise when the cache has been used before it's inited"""
+    def __init__(self):
+        super().__init__("The cache remove error")
+
 class NotFoundError(CacheError):
     """Raise when getting an unsupported store."""
     def __init__(self, store_type, current_type_name):