summaryrefslogtreecommitdiff
path: root/ATRI/plugins/code_runner
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2022-02-03 14:36:24 +0800
committerKyomotoi <[email protected]>2022-02-03 14:36:24 +0800
commit3e32ca3964ff8f40e0b491e87f153040f2348fd0 (patch)
treed04d19ba5a25f6f1f6a9c4f8c398d49eb252df3c /ATRI/plugins/code_runner
parentf5a020d45f7294214bbcd488955b9c391d651a6d (diff)
downloadATRI-3e32ca3964ff8f40e0b491e87f153040f2348fd0.tar.gz
ATRI-3e32ca3964ff8f40e0b491e87f153040f2348fd0.tar.bz2
ATRI-3e32ca3964ff8f40e0b491e87f153040f2348fd0.zip
🔖 更新版本:
更新记录请参考文档: atri.kyomotoi.moe/changelog/overview/
Diffstat (limited to 'ATRI/plugins/code_runner')
-rw-r--r--ATRI/plugins/code_runner/__init__.py9
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))