feat: general接口拦截器

app/general/general_interceptors.py

@@ -0,0 +1,52 @@
+from typing import List
+
+
+class GeneralInterceptor():
+  def __init__(
+    self,
+    module: str,
+    before_list=None,
+    after_list=None,
+    before_insert=None,
+    after_insert=None,
+    before_update=None,
+    after_update=None,
+    before_batch_insert=None,
+    after_batch_insert=None,
+    before_batch_update=None,
+    after_batch_update=None,
+    before_delete=None,
+    after_delete=None,
+  ):
+    self.module = module
+    self.before_list = before_list
+    self.after_list = after_list
+    self.before_insert = before_insert
+    self.after_insert = after_insert
+    self.before_update = before_update
+    self.after_update = after_update
+    self.before_batch_insert = before_batch_insert
+    self.after_batch_insert = after_batch_insert
+    self.before_batch_update = before_batch_update
+    self.after_batch_update = after_batch_update
+    self.before_delete = before_delete
+    self.after_delete = after_delete
+
+
+_general_interceptors: List[GeneralInterceptor] = []
+
+
+def add_general_interceptor(interceptor: GeneralInterceptor):
+  _general_interceptors.append(interceptor)
+
+
+async def invoke_interceptor(method: str, module: str, *args):
+  if module.startswith('/'):
+    module = module[1:]
+  if module.endswith('/'):
+    module = module[:-1]
+  match_interceptors = [item for item in _general_interceptors if item.module == module]
+  for item in match_interceptors:
+    interceptor_method = getattr(item, method, None)  # 使用 getattr 获取方法
+    if interceptor_method and callable(interceptor_method):
+      await interceptor_method(*args)  # 调用方法

app/general/general_utils/sql_service.py

@@ -1,6 +1,7 @@
 import traceback
 
 from app.config.env import env
+from app.general.general_interceptors import invoke_interceptor
 from app.general.general_utils.build_delete_sql import build_delete_sql
 from app.general.general_utils.build_insert_sql import build_insert_sql
 from app.general.general_utils.build_query_sql import build_query_sql
@@ -52,6 +53,8 @@ async def list(session: AsyncSessionDep, query_config, module_config, debug_data
     "orders": get_default_orders(query_config, module_config)
   }
 
+  await invoke_interceptor('before_list', module_config['base'], target_query_config, session)
+
   sql, values = build_query_sql(target_query_config, module_config)
 
   try:
@@ -62,6 +65,8 @@ async def list(session: AsyncSessionDep, query_config, module_config, debug_data
 
     create_convertor(module_config)['decode_list'](result)
 
+    await invoke_interceptor('after_list', module_config['base'], result, session)
+
     if n_only_count:
       return {
         "total": result[0]['total']
@@ -120,6 +125,8 @@ async def insert(session: AsyncSessionDep, query_config, module_config, debug_da
     row_id = (await get_id(session, 1))[0]
     row['id'] = row_id
 
+  await invoke_interceptor('before_insert', module_config['base'], row, session)
+
   try:
     sql, values = build_insert_sql(module_config, row)
 
@@ -134,6 +141,8 @@ async def insert(session: AsyncSessionDep, query_config, module_config, debug_da
 
     item_dict = get_value(result, 'result', None)
 
+    await invoke_interceptor('after_insert', module_config['base'], item_dict, session)
+
     if item_dict is not None:
       return {
         "result": item_dict
@@ -167,6 +176,8 @@ async def batch_insert(session: AsyncSessionDep, query_config, module_config, de
     for index, row in enumerate(rows_without_id):
       row['id'] = new_id_list[index]
 
+  await invoke_interceptor('before_batch_insert', module_config['base'], rows, session)
+
   try:
     for row in rows:
       sql, values = build_insert_sql(module_config, row)
@@ -182,6 +193,9 @@ async def batch_insert(session: AsyncSessionDep, query_config, module_config, de
     result = get_value(result, 'list', [])
 
     if len(result) > 0:
+
+      await invoke_interceptor('after_batch_insert', module_config['base'], result, session)
+
       return {
         "result": result
       }
@@ -218,6 +232,8 @@ async def update(session: AsyncSessionDep, query_config, module_config, debug_da
 
   create_convertor(module_config)['encode_list']([row])
 
+  await invoke_interceptor('before_update', module_config['base'], row, session)
+
   try:
     sql, values = build_update_sql(module_config, row, row.keys() if update_by_fields else None)
 
@@ -232,6 +248,8 @@ async def update(session: AsyncSessionDep, query_config, module_config, debug_da
 
     item_dict = get_value(result, 'result', None)
 
+    await invoke_interceptor('after_update', module_config['base'], item_dict, session)
+
     if item_dict is not None:
       return {
         "result": item_dict
@@ -268,6 +286,8 @@ async def batch_update(session: AsyncSessionDep, query_config, module_config, de
       "rows": rows_without_id,
     }
 
+  await invoke_interceptor('before_batch_update', module_config['base'], rows, session)
+
   try:
     for row in rows:
       sql, values = build_update_sql(module_config, row, row.keys() if update_by_fields else None)
@@ -283,6 +303,9 @@ async def batch_update(session: AsyncSessionDep, query_config, module_config, de
     result = get_value(result, 'list', [])
 
     if len(result) > 0:
+
+      await invoke_interceptor('after_batch_update', module_config['base'], result, session)
+
       return {
         "result": result
       }
@@ -308,6 +331,8 @@ async def delete(session: AsyncSessionDep, query_config, module_config, debug_da
       "error": "id parameter is missing",
     }
 
+  await invoke_interceptor('before_delete', module_config['base'], id, session)
+
   try:
     sql, values = build_delete_sql(module_config, id)
     debug_data.append({"sql": sql, "values": values})
@@ -316,6 +341,9 @@ async def delete(session: AsyncSessionDep, query_config, module_config, debug_da
     deleted_rows = result.rowcount
 
     if deleted_rows >= 1:
+
+      await invoke_interceptor('after_delete', module_config['base'], id, session)
+
       return {"deletedRows": deleted_rows}
     else:
       return {"error": f"delete failed, delete rows is {deleted_rows}", }