添加获取nb配置api

This commit is contained in:
HibiKier 2024-10-16 12:58:01 +08:00
parent fe1634eb2d
commit 792bcaa711
3 changed files with 15 additions and 3 deletions

View File

@ -1,7 +1,9 @@
from datetime import datetime, timedelta from datetime import datetime, timedelta
import nonebot
from nonebot import require from nonebot import require
from fastapi import APIRouter from fastapi import APIRouter
from nonebot.config import Config
from tortoise.functions import Count from tortoise.functions import Count
from tortoise.expressions import RawSQL from tortoise.expressions import RawSQL
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
@ -19,6 +21,8 @@ require("plugin_store")
router = APIRouter(prefix="/dashboard") router = APIRouter(prefix="/dashboard")
driver = nonebot.get_driver()
@router.get( @router.get(
"/get_bot_list", "/get_bot_list",
@ -184,3 +188,14 @@ async def _(query: QueryModel) -> Result[BaseResultModel]:
for v in data: for v in data:
v.connect_time = v.connect_time.replace(tzinfo=None).replace(microsecond=0) v.connect_time = v.connect_time.replace(tzinfo=None).replace(microsecond=0)
return Result.ok(BaseResultModel(total=total, data=data)) return Result.ok(BaseResultModel(total=total, data=data))
@router.get(
"/get_nonebot_config",
dependencies=[authentication()],
response_model=Result[Config],
response_class=JSONResponse,
deprecated="获取nb配置", # type: ignore
)
async def _() -> Result[Config]:
return Result.ok(driver.config)

View File

@ -108,7 +108,6 @@ async def _(bot_id: str | None = None) -> Result[list[BaseInfo]]:
day_call = await Statistics.filter( day_call = await Statistics.filter(
create_time__gte=now - timedelta(hours=now.hour) create_time__gte=now - timedelta(hours=now.hour)
).count() ).count()
select_bot.config = driver.config
select_bot.day_call = day_call select_bot.day_call = day_call
select_bot.connect_count = await BotConnectLog.filter( select_bot.connect_count = await BotConnectLog.filter(
bot_id=select_bot.self_id bot_id=select_bot.self_id

View File

@ -46,8 +46,6 @@ class BaseInfo(BaseModel):
is_select: bool = False is_select: bool = False
"""当前选择""" """当前选择"""
config: Config | None
"""nb配置"""
day_call: int = 0 day_call: int = 0
"""今日调用插件次数""" """今日调用插件次数"""
version: str = "unknown" version: str = "unknown"