feat(api): 新增按聊天流统计消息的API端点

新增了一个API端点 `/messages/stats_by_chat`,用于根据指定的天数统计消息数据。该端点支持按聊天流进行分组,并提供了按用户进一步分组以及过滤机器人自身消息的选项。

主要功能包括:
- 按指定天数范围查询消息。
- 按聊天流(chat_id)聚合统计数据,包括总消息数、发送数和接收数。
- 可选地按用户(user_id)在每个聊天流内进行分组统计。
- 可选地过滤掉机器人自身发送的消息。
This commit is contained in:
minecraft1024a
2025-10-04 12:57:43 +08:00
parent 49025a4973
commit 7d9bfb7fca

View File

@@ -46,3 +46,55 @@ async def get_message_stats(
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@router.get("/messages/stats_by_chat")
async def get_message_stats_by_chat(
days: int = Query(1, ge=1, description="指定查询过去多少天的数据"),
group_by_user: bool = Query(False, description="是否按用户进行分组统计"),
filter_bot: bool = Query(False, description="是否过滤BOT自身的消息")
):
"""
获取BOT在指定天数内按聊天流或按用户统计的消息数据。
"""
try:
end_time = time.time()
start_time = end_time - (days * 24 * 3600)
messages = await message_api.get_messages_by_time(start_time, end_time)
bot_qq = str(global_config.bot.qq_account)
if filter_bot:
messages = [msg for msg in messages if msg.get("user_id") != bot_qq]
stats = {}
for msg in messages:
chat_id = msg.get("chat_id", "unknown")
user_id = msg.get("user_id")
if chat_id not in stats:
stats[chat_id] = {
"total_stats": {"sent": 0, "received": 0, "total": 0},
"user_stats": {}
}
stats[chat_id]["total_stats"]["total"] += 1
if user_id == bot_qq:
stats[chat_id]["total_stats"]["sent"] += 1
else:
stats[chat_id]["total_stats"]["received"] += 1
if group_by_user:
if user_id not in stats[chat_id]["user_stats"]:
stats[chat_id]["user_stats"][user_id] = 0
stats[chat_id]["user_stats"][user_id] += 1
if not group_by_user:
# 如果不按用户分组,则只返回总统计信息
return {chat_id: data["total_stats"] for chat_id, data in stats.items()}
return stats
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))