diff options
Diffstat (limited to 'ATRI/plugins/code_runner')
| -rw-r--r-- | ATRI/plugins/code_runner/__init__.py | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/ATRI/plugins/code_runner/__init__.py b/ATRI/plugins/code_runner/__init__.py index 26f5c52..01b09c6 100644 --- a/ATRI/plugins/code_runner/__init__.py +++ b/ATRI/plugins/code_runner/__init__.py @@ -3,28 +3,24 @@ from random import choice from nonebot.matcher import Matcher from nonebot.params import CommandArg, ArgPlainText from nonebot.adapters.onebot.v11 import MessageEvent, Message, MessageSegment, unescape +from nonebot.adapters.onebot.v11.helpers import Cooldown -from ATRI.utils.limit import FreqLimiter from .data_source import CodeRunner -_flmt = FreqLimiter(5) _flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~"]) code_runner = CodeRunner().on_command("/code", "在线运行一段代码，帮助：/code help") -@code_runner.handle() +@code_runner.handle([Cooldown(5, prompt=_flmt_notice)]) async def _code_runner( matcher: Matcher, event: MessageEvent, args: Message = CommandArg() ): user_id = event.get_user_id() - if not _flmt.check(user_id): - await code_runner.finish(_flmt_notice) - msg = args.extract_plain_text() - print(msg, args, type(msg), type(args)) + if msg: matcher.set_arg("opt", args) else: @@ -44,5 +40,4 @@ async def _(event: MessageEvent, opt: str = ArgPlainText("opt")): else: content = MessageSegment.at(user_id) + await CodeRunner().runner(unescape(opt)) - _flmt.start_cd(user_id) await code_runner.finish(Message(content)) |