From c38bab21b3c85e82250e3b82ebde31e2aca2db92 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Sat, 5 Feb 2022 00:47:52 +0800 Subject: =?UTF-8?q?=F0=9F=9A=91=EF=B8=8F=F0=9F=94=92=EF=B8=8F=F0=9F=94=A7?= =?UTF-8?q?=20=E5=A4=A7=E5=9E=8B=E4=BF=AE=E5=A4=8D=E5=B7=A5=E4=BD=9C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 针对传回消息不同适配、外置限制改为nb内置限制 --- ATRI/plugins/code_runner/__init__.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) (limited to 'ATRI/plugins/code_runner') 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)) -- cgit v1.2.3