nonebot-plugin-group-historian 0.1.0__tar.gz

nonebot_plugin_group_historian-0.1.0/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-group-historian
+Version: 0.1.0
+Summary: 统计每日群聊发言字数 生成话痨榜图片的NoneBot2插件
+Author-email: Wojusensei <3442006415@qq.com>
+Requires-Python: >=3.10
+Requires-Dist: nonebot2>=2.3.0
+Requires-Dist: nonebot-adapter-onebot>=2.0.0
+Requires-Dist: nonebot-plugin-orm[default]>=0.7.0
+Requires-Dist: nonebot-plugin-apscheduler>=0.5.0
+Requires-Dist: nonebot-plugin-localstore>=0.5.0
+Requires-Dist: Pillow>=10.0.0
+Requires-Dist: aiohttp>=3.9.0

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian/__init__.py

@@ -0,0 +1,168 @@
+import asyncio
+from datetime import datetime, timedelta
+from nonebot import on_command, on_message, on_notice, require, get_plugin_config, get_driver
+from nonebot.adapters.onebot.v11 import (
+    Bot,
+    GroupMessageEvent,
+    GroupRecallNoticeEvent,
+    MessageSegment,
+)
+from nonebot.plugin import PluginMetadata
+from nonebot.log import logger
+
+
+# ————————————————————————————
+# 依赖声明
+# ————————————————————————————
+
+require("nonebot_plugin_orm")
+require("nonebot_plugin_apscheduler")
+require("nonebot_plugin_localstore")
+
+
+# ————————————————————————————
+# 元数据 插件身份证
+# ————————————————————————————
+
+__plugin_meta__ = PluginMetadata(
+    name="群聊史官",
+    description="统计每日群聊发言字数 生成话痨榜图片",
+    usage="在群里发送 话痨榜 查看昨日排行\n发送 今日话痨榜 查看今日实时排行\n可加页码 例如 话痨榜 2",
+    type="application",
+    homepage="https://github.com/Wojusensei/nonebot-plugin-group-historian",
+    config=None,
+    supported_adapters={"~onebot.v11"},
+)
+
+
+# ————————————————————————————
+# 导入插件内部模块 放在元数据之后
+# ————————————————————————————
+
+from .config import Config
+from .data import add_message, delete_last_message, get_daily_ranking, clean_old_data
+from .image import create_ranking_image
+
+
+# ————————————————————————————
+# 加载配置 使用 get_plugin_config
+# ————————————————————————————
+
+config = get_plugin_config(Config)
+
+
+# ————————————————————————————
+# 消息缓存 用于撤回时扣除字数
+# key 是消息ID value 是 (群号, QQ号, 字数)
+# ————————————————————————————
+
+message_cache = {}
+
+
+# ————————————————————————————
+# 监听群消息 记录字数
+# ————————————————————————————
+
+msg_handler = on_message(block=False)
+
+
+@msg_handler.handle()
+async def handle_message(event: GroupMessageEvent):
+    text = event.get_plaintext()
+
+    # 过滤纯图片表情等无文字消息
+    if not text or not text.strip():
+        return
+
+    length = len(text.replace(" ", ""))
+    group_id = str(event.group_id)
+    user_id = str(event.user_id)
+    nickname = event.sender.card or event.sender.nickname or user_id
+
+    await add_message(group_id, user_id, nickname, length)
+
+    # 存入缓存 供撤回时扣除
+    message_cache[str(event.message_id)] = (group_id, user_id, length)
+
+
+# ————————————————————————————
+# 监听群撤回 扣除字数
+# ————————————————————————————
+
+recall_handler = on_notice(block=False)
+
+
+@recall_handler.handle()
+async def handle_recall(event: GroupRecallNoticeEvent):
+    msg_id = str(event.message_id)
+    if msg_id in message_cache:
+        group_id, user_id, length = message_cache[msg_id]
+        await delete_last_message(group_id, user_id, length)
+        del message_cache[msg_id]
+
+
+# ————————————————————————————
+# 话痨榜命令
+# 默认查昨日 加今日查今天
+# ————————————————————————————
+
+rank_cmd = on_command("话痨榜", aliases={"今日话痨榜"}, block=True)
+
+
+@rank_cmd.handle()
+async def handle_rank(event: GroupMessageEvent, bot: Bot):
+    raw = event.get_plaintext().strip()
+    parts = raw.split()
+
+    # 判断查今天还是昨天
+    is_today = "今日" in raw
+
+    # 解析页码
+    page = 1
+    for p in parts:
+        if p.isdigit():
+            page = int(p)
+            if page < 1:
+                page = 1
+            break
+
+    # 确定日期
+    date = datetime.now().date() if is_today else (datetime.now().date() - timedelta(days=1))
+
+    # 获取排行榜
+    ranking = await get_daily_ranking(str(event.group_id), date)
+
+    if not ranking:
+        day_text = "今天" if is_today else "昨天"
+        await rank_cmd.finish(f"{day_text}还没有人说话呢", at_sender=True)
+        return
+
+    # 生成图片 放入线程池避免阻塞事件循环
+    rank_count = config.historian_rank_count
+    img_bytes = await asyncio.to_thread(
+        create_ranking_image,
+        ranking,
+        page=page,
+        rank_count=rank_count,
+    )
+
+    await rank_cmd.send(MessageSegment.image(img_bytes))
+
+
+# ————————————————————————————
+# 定时任务 每日凌晨清理旧数据
+# ————————————————————————————
+
+from nonebot_plugin_apscheduler import scheduler
+
+
+@scheduler.scheduled_job("cron", hour=0, minute=0, id="clean_old_historian_data")
+async def scheduled_clean():
+    await clean_old_data(config.historian_data_retention_days)
+
+
+# ————————————————————————————
+# 启动日志
+# ————————————————————————————
+
+logger.info("群聊史官 插件已加载")

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian/config.py

@@ -0,0 +1,14 @@
+from pydantic import BaseModel, Field
+
+
+class Config(BaseModel):
+    """插件配置，用户可在 .env 文件中设置"""
+
+    historian_rank_count: int = Field(
+        default=10,
+        description="排行榜每页显示的人数"
+    )
+    historian_data_retention_days: int = Field(
+        default=30,
+        description="数据保留天数，超过自动清理"
+    )

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian/data.py

@@ -0,0 +1,113 @@
+from datetime import datetime, timedelta
+from sqlalchemy import Column, Integer, String, Date, func
+from nonebot_plugin_orm import Model
+from nonebot import require
+
+require("nonebot_plugin_orm")
+
+
+# ————————————————————————————
+# 每日发言记录表
+# ————————————————————————————
+
+class DailyMessage(Model):
+    __tablename__ = "daily_messages"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    group_id = Column(String, nullable=False)
+    user_id = Column(String, nullable=False)
+    nickname = Column(String)
+    message_length = Column(Integer, default=0)
+    timestamp = Column(Date, nullable=False)
+
+
+# ————————————————————————————
+# 记录
+# ————————————————————————————
+
+async def add_message(group_id: str, user_id: str, nickname: str, length: int):
+    from nonebot_plugin_orm import get_scoped_session
+
+    async with get_scoped_session() as session:
+        record = DailyMessage(
+            group_id=group_id,
+            user_id=user_id,
+            nickname=nickname,
+            message_length=length,
+            timestamp=datetime.now().date(),
+        )
+        session.add(record)
+        await session.commit()
+
+
+# ————————————————————————————
+# 撤回消息时扣除最近一条匹配的字数
+# ————————————————————————————
+
+async def delete_last_message(group_id: str, user_id: str, length: int):
+    from nonebot_plugin_orm import get_scoped_session
+
+    today = datetime.now().date()
+    async with get_scoped_session() as session:
+        record = await session.execute(
+            DailyMessage.__table__.select()
+            .where(
+                DailyMessage.group_id == group_id,
+                DailyMessage.user_id == user_id,
+                DailyMessage.message_length == length,
+                DailyMessage.timestamp == today,
+            )
+            .order_by(DailyMessage.id.desc())
+            .limit(1)
+        )
+        row = record.fetchone()
+        if row:
+            await session.execute(
+                DailyMessage.__table__.delete().where(DailyMessage.id == row.id)
+            )
+            await session.commit()
+
+
+# ————————————————————————————
+# 获取指定日期的字数排行榜
+# 返回 [(user_id, nickname, total), ...]
+# ————————————————————————————
+
+async def get_daily_ranking(group_id: str, date=None) -> list:
+    from nonebot_plugin_orm import get_scoped_session
+
+    if date is None:
+        date = datetime.now().date()
+
+    async with get_scoped_session() as session:
+        result = await session.execute(
+            DailyMessage.__table__.select()
+            .with_only_columns(
+                DailyMessage.user_id,
+                func.max(DailyMessage.nickname).label("nickname"),
+                func.sum(DailyMessage.message_length).label("total"),
+            )
+            .where(
+                DailyMessage.group_id == group_id,
+                DailyMessage.timestamp == date,
+            )
+            .group_by(DailyMessage.user_id)
+            .order_by(func.sum(DailyMessage.message_length).desc())
+        )
+        rows = result.fetchall()
+        return [(row.user_id, row.nickname, row.total) for row in rows]
+
+
+# ————————————————————————————
+# 删除超过保留天数的旧数据
+# ————————————————————————————
+
+async def clean_old_data(retention_days: int):
+    from nonebot_plugin_orm import get_scoped_session
+
+    cutoff = datetime.now().date() - timedelta(days=retention_days)
+    async with get_scoped_session() as session:
+        await session.execute(
+            DailyMessage.__table__.delete().where(DailyMessage.timestamp < cutoff)
+        )
+        await session.commit()

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian/image.py

@@ -0,0 +1,140 @@
+import io
+from PIL import Image, ImageDraw, ImageFont
+from nonebot.log import logger
+from nonebot_plugin_localstore import get_plugin_data_dir
+from nonebot import require
+
+require("nonebot_plugin_localstore")
+
+
+# ————————————————————————————
+# 资源路径
+# 字体和背景图放在插件数据目录下
+# ————————————————————————————
+
+DATA_DIR = get_plugin_data_dir()
+FONT_PATH = DATA_DIR / "font.ttf"
+BG_PATH = DATA_DIR / "background.png"
+
+
+# ————————————————————————————
+# 荣誉图标
+# ————————————————————————————
+
+ICONS = {
+    1: "👑",
+    2: "💎",
+    3: "⭐",
+}
+
+
+# ————————————————————————————
+# 生成排行榜图片
+# ranking: [(user_id, nickname, total), ...]
+# page: 页码1开始
+# rank_count: 每页人数
+# 返回图片的 bytes
+# ————————————————————————————
+
+def create_ranking_image(ranking, page=1, rank_count=10):
+    # ———————— 计算分页 ————————
+    start = (page - 1) * rank_count
+    end = start + rank_count
+    page_data = ranking[start:end]
+
+    # ———————— 画布尺寸 ————————
+    width = 800
+    header_height = 120
+    row_height = 90
+    footer_height = 80
+    height = header_height + row_height * rank_count + footer_height
+
+    # ———————— 背景图 ————————
+    if BG_PATH.exists():
+        bg = Image.open(BG_PATH).convert("RGBA")
+        bg = bg.resize((width, height), Image.LANCZOS)
+        # 蒙灰：降低饱和度并加半透明遮罩
+        bg = bg.convert("L").convert("RGBA")
+        overlay = Image.new("RGBA", (width, height), (0, 0, 0, 140))
+        bg = Image.alpha_composite(bg, overlay)
+    else:
+        bg = Image.new("RGBA", (width, height), (30, 30, 50, 255))
+
+    # ———————— 字体 ————————
+    if FONT_PATH.exists():
+        font_title = ImageFont.truetype(str(FONT_PATH), 36)
+        font_name = ImageFont.truetype(str(FONT_PATH), 24)
+        font_small = ImageFont.truetype(str(FONT_PATH), 18)
+    else:
+        font_title = ImageFont.load_default()
+        font_name = ImageFont.load_default()
+        font_small = ImageFont.load_default()
+
+    # ———————— 创建画布 ————————
+    img = Image.new("RGBA", (width, height), (0, 0, 0, 0))
+    img.paste(bg, (0, 0))
+    draw = ImageDraw.Draw(img)
+
+    # ———————— 标题 ————————
+    title = f"话痨榜 第{page}页"
+    draw.text(
+        (width // 2, 40),
+        title,
+        fill=(255, 215, 0),
+        font=font_title,
+        anchor="ma",
+    )
+
+    # ———————— 分割线 ————————
+    draw.line(
+        (40, header_height - 10, width - 40, header_height - 10),
+        fill=(255, 215, 0, 200),
+        width=2,
+    )
+
+    # ———————— 排行榜条目 ————————
+    for i, (user_id, nickname, total) in enumerate(page_data):
+        rank = start + i + 1
+        y = header_height + i * row_height
+
+        # 排名
+        icon = ICONS.get(rank, "")
+        rank_text = f"{icon} {rank}" if icon else str(rank)
+        draw.text((60, y + row_height // 2), rank_text, fill=(255, 255, 255), font=font_name, anchor="lm")
+
+        # 昵称
+        draw.text((140, y + 25), nickname, fill=(255, 255, 255), font=font_name)
+
+        # QQ号
+        draw.text((140, y + 55), user_id, fill=(180, 180, 180), font=font_small)
+
+        # 字数
+        count_text = f"{total} 字"
+        bbox = draw.textbbox((0, 0), count_text, font=font_name)
+        text_w = bbox[2] - bbox[0]
+        draw.text(
+            (width - 60 - text_w, y + row_height // 2),
+            count_text,
+            fill=(255, 215, 0),
+            font=font_name,
+            anchor="lm",
+        )
+
+    # ———————— 页脚 ————————
+    draw.text(
+        (width // 2, height - 50),
+        "排行由群聊史官自动生成 | 记录昨日话痨数据",
+        fill=(150, 150, 150),
+        font=font_small,
+        anchor="ma",
+    )
+
+    # ———————— 输出 ————————
+    buf = io.BytesIO()
+    img.save(buf, format="PNG")
+    buf.seek(0)
+    return buf.read()
+
+logger.info("群聊史官 图片生成模块已就绪")
+
+#累死了

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian.egg-info/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-group-historian
+Version: 0.1.0
+Summary: 统计每日群聊发言字数 生成话痨榜图片的NoneBot2插件
+Author-email: Wojusensei <3442006415@qq.com>
+Requires-Python: >=3.10
+Requires-Dist: nonebot2>=2.3.0
+Requires-Dist: nonebot-adapter-onebot>=2.0.0
+Requires-Dist: nonebot-plugin-orm[default]>=0.7.0
+Requires-Dist: nonebot-plugin-apscheduler>=0.5.0
+Requires-Dist: nonebot-plugin-localstore>=0.5.0
+Requires-Dist: Pillow>=10.0.0
+Requires-Dist: aiohttp>=3.9.0

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+pyproject.toml
+nonebot_plugin_group_historian/__init__.py
+nonebot_plugin_group_historian/config.py
+nonebot_plugin_group_historian/data.py
+nonebot_plugin_group_historian/image.py
+nonebot_plugin_group_historian.egg-info/PKG-INFO
+nonebot_plugin_group_historian.egg-info/SOURCES.txt
+nonebot_plugin_group_historian.egg-info/dependency_links.txt
+nonebot_plugin_group_historian.egg-info/requires.txt
+nonebot_plugin_group_historian.egg-info/top_level.txt

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian.egg-info/requires.txt

@@ -0,0 +1,7 @@
+nonebot2>=2.3.0
+nonebot-adapter-onebot>=2.0.0
+nonebot-plugin-orm[default]>=0.7.0
+nonebot-plugin-apscheduler>=0.5.0
+nonebot-plugin-localstore>=0.5.0
+Pillow>=10.0.0
+aiohttp>=3.9.0

nonebot_plugin_group_historian-0.1.0/nonebot_plugin_group_historian.egg-info/top_level.txt

@@ -0,0 +1 @@
+nonebot_plugin_group_historian

nonebot_plugin_group_historian-0.1.0/pyproject.toml

@@ -0,0 +1,22 @@
+[project]
+name = "nonebot-plugin-group-historian"
+version = "0.1.0"
+description = "统计每日群聊发言字数 生成话痨榜图片的NoneBot2插件"
+authors = [{name = "Wojusensei", email = "3442006415@qq.com"}]
+requires-python = ">=3.10"
+dependencies = [
+    "nonebot2>=2.3.0",
+    "nonebot-adapter-onebot>=2.0.0",
+    "nonebot-plugin-orm[default]>=0.7.0",
+    "nonebot-plugin-apscheduler>=0.5.0",
+    "nonebot-plugin-localstore>=0.5.0",
+    "Pillow>=10.0.0",
+    "aiohttp>=3.9.0",
+]
+
+[tool.setuptools]
+packages = ["nonebot_plugin_group_historian"]
+
+[build-system]
+requires = ["setuptools>=64.0", "wheel"]
+build-backend = "setuptools.build_meta"

nonebot_plugin_group_historian-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+