88314fda6e
新增了一个API端点 `/messages/stats_by_chat`，用于根据指定的天数统计消息数据。该端点支持按聊天流进行分组，并提供了按用户进一步分组以及过滤机器人自身消息的选项。 主要功能包括： - 按指定天数范围查询消息。 - 按聊天流（chat_id）聚合统计数据，包括总消息数、发送数和接收数。 - 可选地按用户（user_id）在每个聊天流内进行分组统计。 - 可选地过滤掉机器人自身发送的消息。
101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
import time
|
|
from typing import Literal
|
|
|
|
from fastapi import APIRouter, HTTPException, Query
|
|
|
|
from src.config.config import global_config
|
|
from src.plugin_system.apis import message_api
|
|
|
|
router = APIRouter()
|
|
|
|
@router.get("/messages/recent")
|
|
async def get_message_stats(
|
|
days: int = Query(1, ge=1, description="指定查询过去多少天的数据"),
|
|
message_type: Literal["all", "sent", "received"] = Query("all", description="筛选消息类型: 'sent' (BOT发送的), 'received' (BOT接收的), or 'all' (全部)")
|
|
):
|
|
"""
|
|
获取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)
|
|
|
|
sent_count = 0
|
|
received_count = 0
|
|
bot_qq = str(global_config.bot.qq_account)
|
|
|
|
for msg in messages:
|
|
if msg.get("user_id") == bot_qq:
|
|
sent_count += 1
|
|
else:
|
|
received_count += 1
|
|
if message_type == "sent":
|
|
return {"days": days, "message_type": message_type, "count": sent_count}
|
|
elif message_type == "received":
|
|
return {"days": days, "message_type": message_type, "count": received_count}
|
|
else:
|
|
return {
|
|
"days": days,
|
|
"message_type": message_type,
|
|
"sent_count": sent_count,
|
|
"received_count": received_count,
|
|
"total_count": len(messages)
|
|
}
|
|
|
|
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))
|