K2cr2O1
d458413e4b
- 新增 `/status` 指令，展示机器人运行状态和系统指标 - 实现Redis Lua脚本支持原子化计数器操作 - 添加消息收发统计功能 - 完善文档，包括插件开发和性能优化指南 - 重构WebSocket连接池，增加健康检查机制 - 移除旧版编译脚本，优化项目结构
77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
"""
|
||
API 基础模块
|
||
|
||
定义了 API 调用的基础接口和统一处理逻辑。
|
||
"""
|
||
import copy
|
||
from typing import Any, Dict, Optional, TYPE_CHECKING
|
||
|
||
from ..utils.logger import logger
|
||
|
||
if TYPE_CHECKING:
|
||
from ..ws import WS
|
||
|
||
|
||
class BaseAPI:
|
||
"""
|
||
API 基础类，提供了统一的 `call_api` 方法，包含日志记录和异常处理。
|
||
"""
|
||
_ws: "WS"
|
||
self_id: int
|
||
|
||
def __init__(self, ws_client: "WS", self_id: int):
|
||
self._ws = ws_client
|
||
self.self_id = self_id
|
||
|
||
async def call_api(self, action: str, params: Optional[Dict[str, Any]] = None) -> Any:
|
||
"""
|
||
调用 OneBot v11 API，并提供统一的日志和异常处理。
|
||
|
||
:param action: API 动作名称
|
||
:param params: API 参数
|
||
:return: API 响应结果的数据部分
|
||
:raises Exception: 当 API 调用失败或发生网络错误时
|
||
"""
|
||
if params is None:
|
||
params = {}
|
||
|
||
try:
|
||
# 日志记录前，对敏感或过长的参数进行处理
|
||
log_params = copy.deepcopy(params)
|
||
if 'message' in log_params:
|
||
if isinstance(log_params['message'], list):
|
||
for segment in log_params['message']:
|
||
if segment.get('type') == 'image' and 'file' in segment.get('data', {}):
|
||
file_data = segment['data']['file']
|
||
if file_data.startswith('data:image/'):
|
||
segment['data']['file'] = f"{file_data[:50]}... (base64 truncated)"
|
||
elif isinstance(log_params['message'], str) and log_params['message'].startswith('data:image/'):
|
||
log_params['message'] = f"{log_params['message'][:50]}... (base64 truncated)"
|
||
|
||
# 如果是发送消息的动作，则原子化地增加发送消息总数
|
||
if action in ["send_private_msg", "send_group_msg", "send_msg"]:
|
||
from ..managers.redis_manager import redis_manager
|
||
try:
|
||
lua_script = "return redis.call('INCR', KEYS[1])"
|
||
await redis_manager.execute_lua_script(
|
||
script=lua_script,
|
||
keys=["neobot:stats:messages_sent"],
|
||
args=[]
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"发送消息计数失败: {e}")
|
||
|
||
logger.debug(f"调用API -> action: {action}, params: {log_params}")
|
||
response = await self._ws.call_api(action, params)
|
||
logger.debug(f"API响应 <- {response}")
|
||
|
||
if response.get("status") == "failed":
|
||
logger.warning(f"API调用失败: {response}")
|
||
|
||
return response.get("data")
|
||
|
||
except Exception as e:
|
||
logger.error(f"API调用异常: action={action}, params={params}, error={e}")
|
||
raise
|
||
|