From 3e32ca3964ff8f40e0b491e87f153040f2348fd0 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Thu, 3 Feb 2022 14:36:24 +0800 Subject: =?UTF-8?q?=F0=9F=94=96=20=E6=9B=B4=E6=96=B0=E7=89=88=E6=9C=AC:?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 更新记录请参考文档: atri.kyomotoi.moe/changelog/overview/ --- ATRI/plugins/code_runner/__init__.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'ATRI/plugins/code_runner') diff --git a/ATRI/plugins/code_runner/__init__.py b/ATRI/plugins/code_runner/__init__.py index 3f5697b..334c09a 100644 --- a/ATRI/plugins/code_runner/__init__.py +++ b/ATRI/plugins/code_runner/__init__.py @@ -1,7 +1,6 @@ from random import choice -from nonebot.adapters.cqhttp import Bot, MessageEvent -from nonebot.adapters.cqhttp.message import Message, MessageSegment +from nonebot.adapters.onebot.v11 import MessageEvent, Message, MessageSegment, unescape from ATRI.utils.limit import FreqLimiter from .data_source import CodeRunner @@ -15,7 +14,7 @@ code_runner = CodeRunner().on_command("/code", "在线运行一段代码，帮 @code_runner.handle() -async def _code_runner(bot: Bot, event: MessageEvent): +async def _code_runner(event: MessageEvent): user_id = event.get_user_id() if not _flmt.check(user_id): await code_runner.finish(_flmt_notice) @@ -23,14 +22,14 @@ async def _code_runner(bot: Bot, event: MessageEvent): msg = str(event.get_message()) args = msg.split("\n") - if not args: + if not args[0]: content = f"> {MessageSegment.at(user_id)}\n" + "请键入 /code help 以获取帮助~！" elif args[0] == "help": content = f"> {MessageSegment.at(user_id)}\n" + CodeRunner().help() elif args[0] == "list": content = f"> {MessageSegment.at(user_id)}\n" + CodeRunner().list_supp_lang() else: - content = MessageSegment.at(user_id) + await CodeRunner().runner(msg) + content = MessageSegment.at(user_id) + await CodeRunner().runner(unescape(msg)) _flmt.start_cd(user_id) await code_runner.finish(Message(content)) -- cgit v1.2.3