diff options
Diffstat (limited to 'ATRI/plugins/code_runner')
| -rw-r--r-- | ATRI/plugins/code_runner/__init__.py | 9 |
1 files changed, 4 insertions, 5 deletions
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)) |