diff --git a/GenshinUID/__init__.py b/GenshinUID/__init__.py index 6e92aa0c..05e286c4 100644 --- a/GenshinUID/__init__.py +++ b/GenshinUID/__init__.py @@ -1,6 +1,7 @@ from typing import List, Literal, Optional from nonebot.log import logger +from nonebot.adapters import Bot from nonebot.matcher import Matcher from nonebot.permission import SUPERUSER from nonebot.internal.adapter import Event @@ -18,7 +19,7 @@ gsclient: Optional[GsClient] = None @get_message.handle() -async def send_char_adv(ev: Event): +async def send_char_adv(bot: Bot, ev: Event): if gsclient is None: return await connect() @@ -33,6 +34,10 @@ async def send_char_adv(ev: Event): sp_user_type: Optional[ Literal['group', 'direct', 'channel', 'sub_channel'] ] = None + pm = 3 + + if await SUPERUSER(bot, ev): + pm = 2 # qqguild if '_message' in raw_data: @@ -86,6 +91,7 @@ async def send_char_adv(ev: Event): user_id=user_id, content=message, msg_id=msg_id, + user_pm=pm, ) logger.info(f'【发送】[gsuid-core]: {msg.bot_id}') await gsclient._input(msg) @@ -117,3 +123,4 @@ async def connect(): await gsclient.start() except ConnectionRefusedError: logger.error('Core服务器连接失败...请稍后使用[启动core]命令启动...') + logger.error('Core服务器连接失败...请稍后使用[启动core]命令启动...')