mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-01 01:30:04 +08:00
重构webui主页与好友/群组api
This commit is contained in:
@@ -1,75 +1,75 @@
|
||||
from datetime import datetime, timedelta
|
||||
from typing import List, Optional
|
||||
# from datetime import datetime, timedelta
|
||||
# from typing import List, Optional
|
||||
|
||||
import nonebot
|
||||
from fastapi import APIRouter
|
||||
# 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 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
|
||||
# from ..base_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"
|
||||
# AVA_URL = "http://q1.qlogo.cn/g?b=qq&nk={}&s=160"
|
||||
|
||||
router = APIRouter()
|
||||
# router = APIRouter()
|
||||
|
||||
|
||||
@router.get("/get_bot_info", dependencies=[authentication()])
|
||||
async def _(self_id: Optional[str] = None) -> Result:
|
||||
"""
|
||||
获取Bot基础信息
|
||||
# @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.
|
||||
# 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
|
||||
)
|
||||
# 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连接")
|
||||
# return Result.ok(bot_list, "已获取操作列表")
|
||||
# return Result.fail("无Bot连接")
|
||||
|
||||
Reference in New Issue
Block a user