feat✨: 更新内置插件

2026-10-14 16:49:59 +08:00 · 2024-02-25 03:18:34 +08:00
parent 131200a28e
commit eb0572ea77
83 changed files with 7588 additions and 450 deletions
@@ -0,0 +1,83 @@
+from nonebot import on_message
+from nonebot.adapters import Bot
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_alconna import UniMsg
+from nonebot_plugin_apscheduler import scheduler
+from nonebot_plugin_session import EventSession
+
+from zhenxun.configs.config import Config
+from zhenxun.configs.utils import PluginExtraData, RegisterConfig
+from zhenxun.models.chat_history import ChatHistory
+from zhenxun.services.log import logger
+from zhenxun.utils.enum import PluginType
+
+__plugin_meta__ = PluginMetadata(
+    name="消息存储",
+    description="消息存储，被动存储群消息",
+    usage="",
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1",
+        plugin_type=PluginType.HIDDEN,
+        configs=[
+            RegisterConfig(
+                module="chat_history",
+                key="FLAG",
+                value=True,
+                help="是否开启消息自从存储",
+                default_value=True,
+                type=bool,
+            )
+        ],
+    ).dict(),
+)
+
+
+def rule(message: UniMsg) -> bool:
+    return bool(Config.get_config("chat_history", "FLAG") and message)
+
+
+chat_history = on_message(rule=rule, priority=1, block=False)
+
+
+TEMP_LIST = []
+
+
+@chat_history.handle()
+async def _(message: UniMsg, session: EventSession):
+    group_id = session.id3 or session.id2
+    TEMP_LIST.append(
+        ChatHistory(
+            user_id=session.id1,
+            group_id=group_id,
+            text=str(message),
+            plain_text=message.extract_plain_text(),
+            bot_id=session.bot_id,
+            platform=session.platform,
+        )
+    )
+
+
+@scheduler.scheduled_job(
+    "interval",
+    minutes=1,
+)
+async def _():
+    try:
+        message_list = TEMP_LIST.copy()
+        TEMP_LIST.clear()
+        if message_list:
+            await ChatHistory.bulk_create(message_list)
+        logger.debug(f"批量添加聊天记录 {len(message_list)} 条", "定时任务")
+    except Exception as e:
+        logger.error(f"定时批量添加聊天记录", "定时任务", e=e)
+
+
+# @test.handle()
+# async def _(event: MessageEvent):
+#     print(await ChatHistory.get_user_msg(event.user_id, "private"))
+#     print(await ChatHistory.get_user_msg_count(event.user_id, "private"))
+#     print(await ChatHistory.get_user_msg(event.user_id, "group"))
+#     print(await ChatHistory.get_user_msg_count(event.user_id, "group"))
+#     print(await ChatHistory.get_group_msg(event.group_id))
+#     print(await ChatHistory.get_group_msg_count(event.group_id))