feat: general接口拦截器

This commit is contained in:
martsforever
2025-10-15 19:39:13 +08:00
parent b530cb95b7
commit 560c9743dc
7 changed files with 285 additions and 144 deletions
+52
View File
@@ -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) # 调用方法