✨ 智障回复功能

liteyuki/plugins/liteyuki_smart_reply/__init__.py

@@ -0,0 +1,18 @@
+from nonebot.plugin import PluginMetadata
+from .monitors import *
+from .matchers import *
+
+
+__author__ = "snowykami"
+__plugin_meta__ = PluginMetadata(
+    name="轻雪智障回复",
+    description="",
+    usage="",
+    type="application",
+    homepage="https://github.com/snowykami/LiteyukiBot",
+    extra={
+        "liteyuki": True,
+        "toggleable"     : True,
+        "default_enable" : True,
+    }
+)

liteyuki/plugins/liteyuki_smart_reply/matchers.py

@@ -0,0 +1,85 @@
+import asyncio
+import random
+
+from nonebot import Bot, on_message, get_driver, require
+from nonebot.internal.matcher import Matcher
+from nonebot.permission import SUPERUSER
+from nonebot.rule import to_me
+from nonebot.typing import T_State
+
+from liteyuki.utils.base.ly_typing import T_MessageEvent
+from .utils import get_keywords
+from liteyuki.utils.base.word_bank import get_reply
+from liteyuki.utils.event import get_message_type
+from liteyuki.utils.base.permission import GROUP_ADMIN, GROUP_OWNER
+from liteyuki.utils.base.data_manager import group_db, Group
+
+require("nonebot_plugin_alconna")
+from nonebot_plugin_alconna import on_alconna, Alconna, Args, Arparma
+
+nicknames = set()
+driver = get_driver()
+group_reply_probability: dict[str, float] = {
+}
+default_reply_probability = 0.05
+
+
+@on_alconna(
+    Alconna(
+        "set-reply-probability",
+        Args["probability", float, default_reply_probability],
+    ),
+    aliases={"设置回复概率"},
+    permission=SUPERUSER | GROUP_ADMIN | GROUP_OWNER,
+).handle()
+async def _(result: Arparma, event: T_MessageEvent, matcher: Matcher):
+    # 修改内存和数据库的概率值
+    if get_message_type(event) == "group":
+        group_id = event.group_id
+        probability = result.main_args.get("probability")
+        # 保存到内存
+        group_reply_probability[group_id] = probability
+        # 保存到数据库
+        group: Group = group_db.where_one(Group(), "group_id = ?", group_id, default=Group(group_id=str(group_id)))
+        group.config["reply_probability"] = probability
+        group_db.save(group)
+
+        await matcher.send(f"已将群组{group_id}的回复概率设置为{probability}")
+    return
+
+
+@driver.on_bot_connect
+async def _(bot: Bot):
+    global nicknames
+    nicknames.update(bot.config.nickname)
+    # 从数据库加载群组的回复概率
+    groups = group_db.where_all(Group(), default=[])
+    for group in groups:
+        group_reply_probability[group.group_id] = group.config.get("reply_probability", default_reply_probability)
+
+
+@on_message(priority=100).handle()
+async def _(event: T_MessageEvent, bot: Bot, state: T_State, matcher: Matcher):
+    kws = await get_keywords(event.message.extract_plain_text())
+
+    tome = False
+    if await to_me()(event=event, bot=bot, state=state):
+        tome = True
+    else:
+        for kw in kws:
+            if kw in nicknames:
+                tome = True
+                break
+
+    # 回复概率
+    message_type = get_message_type(event)
+    if tome or message_type == "private":
+        p = 1.0
+    else:
+        p = group_reply_probability.get(event.group_id, default_reply_probability)
+
+    if random.random() < p:
+        await asyncio.sleep(random.random() * 2)
+        if reply := get_reply(kws):
+            await matcher.send(reply)
+            return

liteyuki/plugins/liteyuki_smart_reply/utils.py

@@ -0,0 +1,13 @@
+from jieba import lcut
+from nonebot.utils import run_sync
+
+
+@run_sync
+def get_keywords(text: str) -> list[str, ...]:
+    """
+    获取关键词
+    Args:
+        text: 文本
+    Returns:
+    """
+    return lcut(text)