调整为自动注册工具类实例，便于后续扩展。

Author: wenb1n-dev

src/handles/base.py

@@ -1,13 +1,64 @@
-from typing import Dict, Any, Sequence
+from typing import Dict, Any, Sequence, Type, ClassVar
 
 from mcp.types import TextContent, Tool
 
 
-class BaseHandler:
+class ToolRegistry:
+    """工具注册表，用于管理所有工具实例"""
+    _tools: ClassVar[Dict[str, 'BaseHandler']] = {}
+
+    @classmethod
+    def register(cls, tool_class: Type['BaseHandler']) -> Type['BaseHandler']:
+        """注册工具类
+        
+        Args:
+            tool_class: 要注册的工具类
+            
+        Returns:
+            返回注册的工具类，方便作为装饰器使用
+        """
+        tool = tool_class()
+        cls._tools[tool.name] = tool
+        return tool_class
+
+    @classmethod
+    def get_tool(cls, name: str) -> 'BaseHandler':
+        """获取工具实例
+        
+        Args:
+            name: 工具名称
+            
+        Returns:
+            工具实例
+            
+        Raises:
+            ValueError: 当工具不存在时抛出
+        """
+        if name not in cls._tools:
+            raise ValueError(f"未知的工具: {name}")
+        return cls._tools[name]
+
+    @classmethod
+    def get_all_tools(cls) -> list[Tool]:
+        """获取所有工具的描述
+        
+        Returns:
+            所有工具的描述列表
+        """
+        return [tool.get_tool_description() for tool in cls._tools.values()]
 
+
+class BaseHandler:
+    """工具基类"""
     name: str = ""
     description: str = ""
 
+    def __init_subclass__(cls, **kwargs):
+        """子类初始化时自动注册到工具注册表"""
+        super().__init_subclass__(**kwargs)
+        if cls.name:  # 只注册有名称的工具
+            ToolRegistry.register(cls)
+
     def get_tool_description(self) -> Tool:
         raise NotImplementedError
 

src/server.py

@@ -10,49 +10,18 @@
 from starlette.applications import Starlette
 from starlette.routing import Route, Mount
 
-from handles import (
-    ExecuteSQL,
-    GetChineseInitials,
-    GetTableIndex,
-    GetTableLock,
-    GetTableName,
-    GetTableDesc
-)
-
-# 初始化工具实例
-execute_sql = ExecuteSQL()
-get_chinese_initials = GetChineseInitials()
-get_table_index = GetTableIndex()
-get_table_desc = GetTableDesc()
-get_table_name = GetTableName()
-get_table_lock = GetTableLock()
-
+from handles.base import ToolRegistry
 
 # 初始化服务器
 app = Server("operateMysql")
 
 
 @app.list_tools()
 async def list_tools() -> list[Tool]:
-    """列出所有可用的MySQL操作工具
-    
-    Returns:
-        list[Tool]: 返回工具列表，包含:
-            - execute_sql: 执行SQL语句
-            - get_chinese_initials: 获取中文拼音首字母
-            - get_table_index: 获取表索引信息
-            - get_table_desc: 获取表结构描述
-            - get_table_name: 获取表名
-            - get_table_lock: 获取表锁信息
     """
-    return [
-        execute_sql.get_tool_description(),
-        get_chinese_initials.get_tool_description(),
-        get_table_index.get_tool_description(),
-        get_table_desc.get_tool_description(),
-        get_table_name.get_tool_description(),
-        get_table_lock.get_tool_description()
-    ]
+        列出所有可用的MySQL操作工具
+    """
+    return ToolRegistry.get_all_tools()
 
 @app.call_tool()
 async def call_tool(name: str, arguments: dict) -> Sequence[TextContent]:
@@ -68,20 +37,9 @@ async def call_tool(name: str, arguments: dict) -> Sequence[TextContent]:
     Raises:
         ValueError: 当指定了未知的工具名称时抛出异常
     """
-    if name == execute_sql.name:
-        return await execute_sql.run_tool(arguments)
-    elif name == get_table_index.name:
-        return await get_table_index.run_tool(arguments)
-    elif name == get_table_name.name:
-        return await get_table_name.run_tool(arguments)
-    elif name == get_table_lock.name:
-        return await get_table_lock.run_tool(arguments)
-    elif name == get_table_desc.name:
-        return await get_table_desc.run_tool(arguments)
-    elif name == get_chinese_initials.name:
-        return await get_chinese_initials.run_tool(arguments)
-
-    raise ValueError(f"未知的工具: {name}")
+    tool = ToolRegistry.get_tool(name)
+
+    return await tool.run_tool(arguments)
 
 
 async def run_stdio():