feat✨: 更新内置插件

zhenxun/builtin_plugins/chat_history/__init__.py

@@ -0,0 +1,4 @@
+import nonebot
+from pathlib import Path
+
+nonebot.load_plugins(str(Path(__file__).parent.resolve()))

zhenxun/builtin_plugins/chat_history/chat_message.py

@@ -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))

zhenxun/builtin_plugins/chat_history/chat_message_handle.py

@@ -0,0 +1,116 @@
+from datetime import datetime, timedelta
+
+import pytz
+from nonebot.adapters import Bot
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_alconna import (
+    Alconna,
+    Args,
+    Arparma,
+    Match,
+    Option,
+    on_alconna,
+    store_true,
+)
+from nonebot_plugin_saa import Image, Text
+from nonebot_plugin_session import EventSession
+
+from zhenxun.configs.utils import PluginExtraData
+from zhenxun.models.chat_history import ChatHistory
+from zhenxun.models.group_member_info import GroupInfoUser
+from zhenxun.services.log import logger
+from zhenxun.utils.enum import PluginType
+from zhenxun.utils.image_utils import ImageTemplate
+
+__plugin_meta__ = PluginMetadata(
+    name="消息统计查询",
+    description="消息统计查询",
+    usage="",
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1",
+        plugin_type=PluginType.NORMAL,
+        menu_type="数据统计",
+    ).dict(),
+)
+
+# TODO: shortcut
+
+_matcher = on_alconna(
+    Alconna(
+        "消息排行",
+        Option("--des", default=False, action=store_true),
+        Args["type?", ["日", "周", "月", "年"]]["count?", int, 10],
+    ),
+    priority=5,
+    block=True,
+)
+
+
+@_matcher.handle()
+async def _(
+    bot: Bot,
+    session: EventSession,
+    arparma: Arparma,
+    type: Match[str],
+    count: Match[int],
+):
+    group_id = session.id3 or session.id2
+    if not group_id:
+        await Text("群组id为空...").finish()
+    time_now = datetime.now()
+    date_scope = None
+    zero_today = time_now - timedelta(
+        hours=time_now.hour, minutes=time_now.minute, seconds=time_now.second
+    )
+    date = type.result if type.available else None
+    if date:
+        if date in ["日"]:
+            date_scope = (zero_today, time_now)
+        elif date in ["周"]:
+            date_scope = (time_now - timedelta(days=7), time_now)
+        elif date in ["月"]:
+            date_scope = (time_now - timedelta(days=30), time_now)
+    column_name = ["名次", "昵称", "发言次数"]
+    if rank_data := await ChatHistory.get_group_msg_rank(
+        group_id, count.result, "DES" if arparma.find("des") else "DESC", date_scope
+    ):
+        idx = 1
+        data_list = []
+        for uid, num in rank_data:
+            if user := await GroupInfoUser.filter(
+                user_id=uid, group_id=group_id
+            ).first():
+                user_name = user.user_name
+            else:
+                user_name = uid
+            data_list.append([idx, user_name, num])
+            idx += 1
+        if not date_scope:
+            if date_scope := await ChatHistory.get_group_first_msg_datetime(group_id):
+                date_scope = date_scope.astimezone(
+                    pytz.timezone("Asia/Shanghai")
+                ).replace(microsecond=0)
+            else:
+                date_scope = time_now.replace(microsecond=0)
+            date_str = f"{date_scope} - 至今"
+        else:
+            date_str = f"{date_scope[0].replace(microsecond=0)} - {date_scope[1].replace(microsecond=0)}"
+        A = await ImageTemplate.table_page(
+            f"消息排行({count.result})", date_str, column_name, data_list
+        )
+        logger.info(
+            f"查看消息排行 数量={count.result}", arparma.header_result, session=session
+        )
+        await Image(A.pic2bs4()).finish(reply=True)
+    await Text("群组消息记录为空...").finish()
+
+
+# # @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))