mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
from datetime import datetime, timedelta
|
|
from typing import List, Optional
|
|
|
|
import nonebot
|
|
from fastapi import APIRouter
|
|
|
|
from configs.config import Config
|
|
from models.chat_history import ChatHistory
|
|
from services.log import logger
|
|
from utils.manager import plugin_data_manager, plugins2settings_manager, plugins_manager
|
|
from utils.manager.models import PluginData, PluginType
|
|
|
|
from ..models.model import BotInfo, Result
|
|
from ..models.params import UpdateConfig, UpdatePlugin
|
|
from ..utils import authentication
|
|
|
|
AVA_URL = "http://q1.qlogo.cn/g?b=qq&nk={}&s=160"
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/get_bot_info", dependencies=[authentication()])
|
|
async def _(self_id: Optional[str] = None) -> Result:
|
|
"""
|
|
获取Bot基础信息
|
|
|
|
Args:
|
|
qq (Optional[str], optional): qq号. Defaults to None.
|
|
|
|
Returns:
|
|
Result: 获取指定bot信息与bot列表
|
|
"""
|
|
bot_list: List[BotInfo] = []
|
|
if bots := nonebot.get_bots():
|
|
select_bot: BotInfo
|
|
for key, bot in bots.items():
|
|
bot_list.append(
|
|
BotInfo(
|
|
bot=bot, # type: ignore
|
|
self_id=bot.self_id,
|
|
nickname="可爱的小真寻",
|
|
ava_url=AVA_URL.format(bot.self_id),
|
|
)
|
|
)
|
|
if _bl := [b for b in bot_list if b.self_id == self_id]:
|
|
select_bot = _bl[0]
|
|
else:
|
|
select_bot = bot_list[0]
|
|
select_bot.is_select = True
|
|
now = datetime.now()
|
|
select_bot.received_messages = await ChatHistory.filter(
|
|
bot_id=int(select_bot.self_id)
|
|
).count()
|
|
select_bot.received_messages_day = await ChatHistory.filter(
|
|
bot_id=int(select_bot.self_id),
|
|
create_time__gte=now - timedelta(hours=now.hour),
|
|
).count()
|
|
select_bot.received_messages_week = await ChatHistory.filter(
|
|
bot_id=int(select_bot.self_id),
|
|
create_time__gte=now - timedelta(days=7),
|
|
).count()
|
|
select_bot.group_count = len(await select_bot.bot.get_group_list())
|
|
select_bot.friend_count = len(await select_bot.bot.get_friend_list())
|
|
for bot in bot_list:
|
|
bot.bot = None # type: ignore
|
|
# 插件加载数量
|
|
select_bot.plugin_count = len(plugins2settings_manager)
|
|
pm_data = plugins_manager.get_data()
|
|
select_bot.fail_plugin_count = len([pd for pd in pm_data if pm_data[pd].error])
|
|
select_bot.success_plugin_count = (
|
|
select_bot.plugin_count - select_bot.fail_plugin_count
|
|
)
|
|
|
|
return Result.ok(bot_list, "已获取操作列表")
|
|
return Result.fail("无Bot连接")
|