feat✨: 添加词条word_bank

2026-10-22 04:30:05 +08:00 · 2024-07-27 04:30:03 +08:00
parent cf208e2f64
commit 137870b698
11 changed files with 1371 additions and 9 deletions
@@ -0,0 +1,59 @@
+from io import BytesIO
+
+import imagehash
+from nonebot.adapters import Bot, Event
+from nonebot.typing import T_State
+from nonebot_plugin_alconna import At as alcAt
+from nonebot_plugin_alconna import Text as alcText
+from nonebot_plugin_alconna import UniMsg
+from nonebot_plugin_session import EventSession
+from PIL import Image
+from requests import session
+
+from zhenxun.services.log import logger
+from zhenxun.utils.http_utils import AsyncHttpx
+
+from ._data_source import get_img_and_at_list
+from ._model import WordBank
+
+
+async def check(
+    bot: Bot,
+    event: Event,
+    message: UniMsg,
+    session: EventSession,
+    state: T_State,
+) -> bool:
+    text = message.extract_plain_text().strip()
+    img_list, at_list = get_img_and_at_list(message)
+    problem = text
+    if not text and len(img_list) == 1:
+        try:
+            r = await AsyncHttpx.get(img_list[0])
+            problem = str(imagehash.average_hash(Image.open(BytesIO(r.content))))
+        except Exception as e:
+            logger.warning(f"获取图片失败", "词条检测", session=session, e=e)
+    if at_list:
+        temp = ""
+        # TODO: 支持更多消息类型
+        for msg in message:
+            if isinstance(msg, alcAt):
+                temp += f"[at:{msg.target}]"
+            elif isinstance(msg, alcText):
+                temp += msg.text
+        problem = temp
+    if event.is_tome() and bot.config.nickname:
+        if isinstance(message[0], alcAt) and message[0].target == bot.self_id:
+            problem = f"[at:{bot.self_id}]" + problem
+        else:
+            if problem and bot.config.nickname:
+                nickname = [
+                    nk for nk in bot.config.nickname if str(message).startswith(nk)
+                ]
+                problem = nickname[0] + problem if nickname else problem
+    if problem and (
+        await WordBank.check_problem(session.id3 or session.id2, problem) is not None
+    ):
+        state["problem"] = problem  # type: ignore
+        return True
+    return False