From d36e75c4f7d0a03b36ac84cb80beed97aca4dd76 Mon Sep 17 00:00:00 2001 From: suyiiyii Date: Mon, 1 Jul 2024 11:49:06 +0000 Subject: [PATCH] =?UTF-8?q?:bug:=20=E8=8E=B7=E5=8F=96=E5=90=8E=E5=8F=B0?= =?UTF-8?q?=E9=93=BE=E6=8E=A5=E7=9A=84=E6=97=B6=E5=80=99=EF=BC=8C=E5=AF=B9?= =?UTF-8?q?=E6=B2=A1=E6=9C=89=E6=9D=83=E9=99=90=E7=9A=84=E7=94=A8=E6=88=B7?= =?UTF-8?q?=E8=BF=9B=E8=A1=8C=E6=8F=90=E7=A4=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nonebot_bison/admin_page/__init__.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/nonebot_bison/admin_page/__init__.py b/nonebot_bison/admin_page/__init__.py index 4c2d4e6..3e3505c 100644 --- a/nonebot_bison/admin_page/__init__.py +++ b/nonebot_bison/admin_page/__init__.py @@ -2,6 +2,7 @@ import os from pathlib import Path from typing import TYPE_CHECKING +import nonebot from nonebot.log import logger from nonebot.rule import to_me from nonebot.typing import T_State @@ -9,6 +10,7 @@ from nonebot import get_driver, on_command from nonebot.adapters.onebot.v11 import Bot from nonebot.adapters.onebot.v11.event import PrivateMessageEvent +from .api import get_admin_groups from .api import router as api_router from ..plugin_config import plugin_config from .token_manager import token_manager as tm @@ -65,6 +67,11 @@ def register_get_token_handler(): @get_token.handle() async def send_token(bot: "Bot", event: PrivateMessageEvent, state: T_State): + qq = event.sender.user_id + if qq and str(qq) not in nonebot.get_driver().config.superusers and len(await get_admin_groups(int(qq))) == 0: + await get_token.finish("没有可以管理的群组") + return + token = tm.get_user_token((event.get_user_id(), event.sender.nickname)) await get_token.finish(f"请访问: {plugin_config.outer_url / 'auth' / token}")