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/rich/__init__.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) (limited to 'ATRI/plugins/rich') diff --git a/ATRI/plugins/rich/__init__.py b/ATRI/plugins/rich/__init__.py index 3b49750..2d8f8c6 100644 --- a/ATRI/plugins/rich/__init__.py +++ b/ATRI/plugins/rich/__init__.py @@ -1,20 +1,15 @@ from nonebot.adapters.onebot.v11 import MessageEvent +from nonebot.adapters.onebot.v11.helpers import Cooldown -from ATRI.utils.limit import FreqLimiter from ATRI.log import logger as log from .data_source import Rich -_rich_flmt = FreqLimiter(3) bili_rich = Rich().on_message("小程序检测", "小程序爪巴", block=False) -@bili_rich.handle() +@bili_rich.handle([Cooldown(3)]) async def _fk_bili(event: MessageEvent): - user_id = event.get_user_id() - if not _rich_flmt.check(user_id): - return - msg = str(event.message) try: result, is_ok = await Rich().fk_bili(msg) @@ -23,5 +18,4 @@ async def _fk_bili(event: MessageEvent): log.debug(result, is_ok) if not is_ok: return - _rich_flmt.start_cd(user_id) await bili_rich.finish(result) -- cgit v1.2.3