nonebot-plugin-group-heat 0.1.0__tar.gz

nonebot_plugin_group_heat-0.1.0/PKG-INFO

@@ -0,0 +1,62 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-group-heat
+Version: 0.1.0
+Summary: 群热度统计插件 - 实时查询群内热度，生成昨日热度折线图
+Author-email: Wojusensei <3442006415@qq.com>
+License: MIT
+Project-URL: Homepage, https://github.com/Wojusensei/nonebot-plugin-group-heat
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: nonebot2>=2.3.0
+Requires-Dist: nonebot-adapter-onebot>=2.0.0
+Requires-Dist: nonebot-plugin-localstore>=0.5.0
+Requires-Dist: matplotlib>=3.5.0
+Requires-Dist: numpy>=1.21.0
+
+# nonebot-plugin-group-heat
+
+群热度统计插件 for NoneBot2
+
+## 功能
+
+- `/群热度` - 获取过去30分钟的群热度值，附带 bot 评价
+- `/昨日热度图` - 生成昨日每30分钟的热度折线图，并显示平均热度
+
+## 热度计算规则
+
+| 消息类型 | 热度值 |
+|---------|--------|
+| 文本消息 | +0.05 |
+| 表情包 | +0.20 |
+| 文件（图片/视频/语音等） | +0.30 |
+| 其他消息 | 不计入 |
+
+- 每个统计周期的初始热度为 -10°
+
+## 评价机制
+
+| 热度范围 | 评价 |
+|---------|------|
+| < 0 | 群成冰块啦，群主快开暖气 |
+| 0 ~ 10 | 是冬天到了吗，好冷www |
+| 10 ~ 20 | 温度非常舒适，大家继续努力~ |
+| 20 ~ 30 | 群热度达到最佳状态！ |
+| 30 ~ 39 | 好热，群主快开空调！ |
+| ≥ 39 | 请发送高温补贴喵。。 |
+
+## 安装
+
+```bash
+nb plugin install nonebot-plugin-group-heat
+pip install nonebot-plugin-group-heat(也可以)
+
+## 使用
+在群聊中发送以下命令：
+
+/群热度 - 查询过去30分钟的群热度
+
+/昨日热度图 - 获取昨日热度折线图
+
+## 开源协议
+MIT

nonebot_plugin_group_heat-0.1.0/README.md

@@ -0,0 +1,46 @@
+# nonebot-plugin-group-heat
+
+群热度统计插件 for NoneBot2
+
+## 功能
+
+- `/群热度` - 获取过去30分钟的群热度值，附带 bot 评价
+- `/昨日热度图` - 生成昨日每30分钟的热度折线图，并显示平均热度
+
+## 热度计算规则
+
+| 消息类型 | 热度值 |
+|---------|--------|
+| 文本消息 | +0.05 |
+| 表情包 | +0.20 |
+| 文件（图片/视频/语音等） | +0.30 |
+| 其他消息 | 不计入 |
+
+- 每个统计周期的初始热度为 -10°
+
+## 评价机制
+
+| 热度范围 | 评价 |
+|---------|------|
+| < 0 | 群成冰块啦，群主快开暖气 |
+| 0 ~ 10 | 是冬天到了吗，好冷www |
+| 10 ~ 20 | 温度非常舒适，大家继续努力~ |
+| 20 ~ 30 | 群热度达到最佳状态！ |
+| 30 ~ 39 | 好热，群主快开空调！ |
+| ≥ 39 | 请发送高温补贴喵。。 |
+
+## 安装
+
+```bash
+nb plugin install nonebot-plugin-group-heat
+pip install nonebot-plugin-group-heat(也可以)
+
+## 使用
+在群聊中发送以下命令：
+
+/群热度 - 查询过去30分钟的群热度
+
+/昨日热度图 - 获取昨日热度折线图
+
+## 开源协议
+MIT

nonebot_plugin_group_heat-0.1.0/pyproject.toml

@@ -0,0 +1,29 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "nonebot-plugin-group-heat"
+version = "0.1.0"
+description = "群热度统计插件 - 实时查询群内热度，生成昨日热度折线图"
+authors = [{name = "Wojusensei", email = "3442006415@qq.com"}]
+readme = "README.md"
+license = {text = "MIT"}
+classifiers = [
+    "Programming Language :: Python :: 3",
+]
+dependencies = [
+    "nonebot2>=2.3.0",
+    "nonebot-adapter-onebot>=2.0.0",
+    "nonebot-plugin-localstore>=0.5.0",
+    "matplotlib>=3.5.0",
+    "numpy>=1.21.0",
+]
+requires-python = ">=3.9"
+
+[project.urls]
+Homepage = "https://github.com/Wojusensei/nonebot-plugin-group-heat"
+
+[tool.setuptools]
+packages = ["nonebot_plugin_group_heat"]
+package-dir = {"" = "src"}

nonebot_plugin_group_heat-0.1.0/setup.cfg

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

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat/__init__.py

@@ -0,0 +1,128 @@
+from nonebot.plugin import PluginMetadata
+from nonebot.adapters.onebot.v11 import Event, Bot, MessageSegment
+from nonebot.adapters.onebot.v11.event import GroupMessageEvent
+from .config import Config
+
+
+__plugin_meta__ = PluginMetadata(
+    name="群热度统计",
+    description="统计群内热度，支持实时查询和昨日热度图",
+    usage=(
+        "/群热度 - 获取过去30分钟的群热度\n"
+        "/昨日热度图 - 获取昨日每30分钟的热度折线图和平均热度"
+    ),
+    type="application",
+    homepage="https://github.com/Wojusensei/nonebot-plugin-group-heat",
+    config=Config,
+    supported_adapters={"~onebot.v11"},
+)
+
+
+# 延迟导入其他需要 NoneBot 初始化的模块
+_imported = False
+
+
+def _import_all():
+    global _imported
+    if _imported:
+        return
+    global re, datetime, on_command, on_message, get_driver, CommandArg, logger
+    global init_db, add_message, get_recent_heat, get_yesterday_heat
+    global draw_heat_line, get_heat_comment
+    
+    import re
+    from datetime import datetime
+    from nonebot import on_command, on_message, get_driver
+    from nonebot.params import CommandArg
+    from nonebot.log import logger
+    
+    from .database import init_db, add_message, get_recent_heat, get_yesterday_heat
+    from .heat_image import draw_heat_line, get_heat_comment
+    
+    _imported = True
+
+
+async def get_message_type(event: Event) -> str:
+    msg = event.get_message()
+    for seg in msg:
+        if seg.type == "text":
+            return "text"
+        elif seg.type == "face":
+            return "sticker"
+        elif seg.type in ["file", "image", "record", "video"]:
+            return "file"
+    return "other"
+
+
+try:
+    _import_all()
+    driver = get_driver()
+    
+    @driver.on_startup
+    async def startup():
+        await init_db()
+        logger.info("群热度插件数据库初始化完成")
+except ValueError:
+    pass
+
+
+try:
+    _import_all()
+    msg_recorder = on_message(priority=1, block=False)
+    
+    @msg_recorder.handle()
+    async def record_message(bot: Bot, event: Event):
+        if not isinstance(event, GroupMessageEvent):
+            return
+        group_id = event.group_id
+        user_id = event.user_id
+        msg_type = await get_message_type(event)
+        timestamp = datetime.now().timestamp()
+        await add_message(group_id, user_id, msg_type, timestamp)
+except ValueError:
+    pass
+
+
+try:
+    _import_all()
+    heat_cmd = on_command("/群热度", aliases={"群热度"}, priority=10, block=True)
+    
+    @heat_cmd.handle()
+    async def handle_heat(event: Event):
+        if not isinstance(event, GroupMessageEvent):
+            await heat_cmd.finish("该命令仅支持群聊")
+        group_id = event.group_id
+        try:
+            heat = await get_recent_heat(group_id, minutes=30)
+            comment = get_heat_comment(heat)
+            await heat_cmd.finish(f"过去30分钟的群热度：{heat:.2f}°\n{comment}")
+        except Exception as e:
+            logger.error(f"获取群热度失败: {e}")
+            await heat_cmd.finish("获取热度失败，请稍后再试")
+except ValueError:
+    pass
+
+
+try:
+    _import_all()
+    yesterday_cmd = on_command("/昨日热度图", aliases={"昨日热度图"}, priority=10, block=True)
+    
+    @yesterday_cmd.handle()
+    async def handle_yesterday(event: Event):
+        if not isinstance(event, GroupMessageEvent):
+            await yesterday_cmd.finish("该命令仅支持群聊")
+        group_id = event.group_id
+        try:
+            heat_values, time_labels, avg_heat = await get_yesterday_heat(group_id)
+            if not heat_values:
+                await yesterday_cmd.finish("暂无昨日数据，请明天再来~")
+            img_path = draw_heat_line(heat_values, time_labels, avg_heat)
+            comment = get_heat_comment(avg_heat)
+            msg = f"昨日群平均热度：{avg_heat:.2f}°\n{comment}"
+            await yesterday_cmd.send(MessageSegment.image(img_path))
+            await yesterday_cmd.finish(msg)
+        except Exception as e:
+            logger.error(f"生成昨日热度图失败: {e}")
+            await yesterday_cmd.finish("生成热度图失败，请稍后再试")
+except ValueError:
+    pass

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat/config.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+
+class Config(BaseModel):
+    """Plugin Config"""
+    # 无需配置，即装即用
+    pass

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat/database.py

@@ -0,0 +1,116 @@
+import sqlite3
+import asyncio
+from pathlib import Path
+from datetime import datetime, timedelta
+from typing import List, Tuple
+
+
+def get_data_dir():
+    from nonebot_plugin_localstore import get_data_dir as _get_data_dir
+    return _get_data_dir("nonebot_plugin_group_heat")
+
+
+def get_db_path():
+    data_dir = get_data_dir()
+    data_dir.mkdir(parents=True, exist_ok=True)
+    return data_dir / "heat.db"
+
+
+async def init_db():
+    def _init():
+        conn = sqlite3.connect(str(get_db_path()))
+        c = conn.cursor()
+        c.execute('''
+            CREATE TABLE IF NOT EXISTS messages (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                group_id INTEGER NOT NULL,
+                user_id INTEGER NOT NULL,
+                msg_type TEXT NOT NULL,
+                timestamp REAL NOT NULL
+            )
+        ''')
+        c.execute('CREATE INDEX IF NOT EXISTS idx_group_time ON messages (group_id, timestamp)')
+        conn.commit()
+        conn.close()
+    await asyncio.get_event_loop().run_in_executor(None, _init)
+
+
+async def add_message(group_id: int, user_id: int, msg_type: str, timestamp: float):
+    def _add():
+        conn = sqlite3.connect(str(get_db_path()))
+        c = conn.cursor()
+        c.execute(
+            "INSERT INTO messages (group_id, user_id, msg_type, timestamp) VALUES (?, ?, ?, ?)",
+            (group_id, user_id, msg_type, timestamp)
+        )
+        conn.commit()
+        conn.close()
+    await asyncio.get_event_loop().run_in_executor(None, _add)
+
+
+async def get_recent_heat(group_id: int, minutes: int = 30) -> float:
+    now = datetime.now().timestamp()
+    start = now - minutes * 60
+
+    def _calc():
+        conn = sqlite3.connect(str(get_db_path()))
+        c = conn.cursor()
+        c.execute('SELECT msg_type FROM messages WHERE group_id = ? AND timestamp >= ?', (group_id, start))
+        rows = c.fetchall()
+        conn.close()
+        heat = -10.0
+        for row in rows:
+            msg_type = row[0]
+            if msg_type == 'text':
+                heat += 0.05
+            elif msg_type == 'sticker':
+                heat += 0.2
+            elif msg_type == 'file':
+                heat += 0.3
+        return heat
+    return await asyncio.get_event_loop().run_in_executor(None, _calc)
+
+
+async def get_yesterday_heat(group_id: int) -> Tuple[List[float], List[str], float]:
+    now = datetime.now()
+    yesterday_start = now.replace(hour=0, minute=0, second=0, microsecond=0) - timedelta(days=1)
+    yesterday_end = yesterday_start + timedelta(days=1)
+
+    intervals = []
+    current = yesterday_start
+    while current < yesterday_end:
+        intervals.append(current)
+        current += timedelta(minutes=30)
+
+    heat_values = []
+    time_labels = []
+
+    for interval_start in intervals:
+        interval_end = interval_start + timedelta(minutes=30)
+        start_ts = interval_start.timestamp()
+        end_ts = interval_end.timestamp()
+
+        def _calc_interval():
+            conn = sqlite3.connect(str(get_db_path()))
+            c = conn.cursor()
+            c.execute('SELECT msg_type FROM messages WHERE group_id = ? AND timestamp >= ? AND timestamp < ?',
+                      (group_id, start_ts, end_ts))
+            rows = c.fetchall()
+            conn.close()
+            heat = -10.0
+            for row in rows:
+                t = row[0]
+                if t == 'text':
+                    heat += 0.05
+                elif t == 'sticker':
+                    heat += 0.2
+                elif t == 'file':
+                    heat += 0.3
+            return heat
+
+        heat = await asyncio.get_event_loop().run_in_executor(None, _calc_interval)
+        heat_values.append(heat)
+        time_labels.append(interval_start.strftime("%H:%M"))
+
+    avg_heat = sum(heat_values) / len(heat_values) if heat_values else -10.0
+    return heat_values, time_labels, avg_heat

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat/heat_image.py

@@ -0,0 +1,58 @@
+import matplotlib
+matplotlib.use('Agg')
+import matplotlib.pyplot as plt
+import numpy as np
+from pathlib import Path
+from typing import List
+
+
+def get_cache_dir():
+    from nonebot_plugin_localstore import get_cache_dir as _get_cache_dir
+    cache_dir = _get_cache_dir("nonebot_plugin_group_heat")
+    cache_dir.mkdir(parents=True, exist_ok=True)
+    return cache_dir
+
+
+def draw_heat_line(heat_values: List[float], time_labels: List[str], avg_heat: float) -> Path:
+    plt.rcParams['font.sans-serif'] = ['Microsoft YaHei', 'SimHei']
+    plt.rcParams['axes.unicode_minus'] = False
+
+    fig, ax = plt.subplots(figsize=(14, 6))
+    x = np.arange(len(heat_values))
+
+    ax.plot(x, heat_values, marker='o', linestyle='-', color='#FF6B6B', linewidth=2, markersize=6)
+    ax.fill_between(x, heat_values, -10, alpha=0.2, color='orange')
+    ax.axhline(y=avg_heat, color='blue', linestyle='--', linewidth=1.5, label=f'平均热度: {avg_heat:.2f}')
+
+    ax.set_xlabel('时间 (每30分钟)', fontsize=12)
+    ax.set_ylabel('热度值', fontsize=12)
+    ax.set_title('昨日群热度变化趋势图', fontsize=14, fontweight='bold')
+
+    tick_step = max(1, len(time_labels) // 12)
+    ax.set_xticks(x[::tick_step])
+    ax.set_xticklabels(time_labels[::tick_step], rotation=45, ha='right', fontsize=8)
+
+    ax.grid(True, alpha=0.3)
+    ax.legend(loc='upper right')
+    plt.tight_layout()
+
+    img_path = get_cache_dir() / "yesterday_heat.png"
+    plt.savefig(img_path, dpi=100, bbox_inches='tight')
+    plt.close(fig)
+
+    return img_path
+
+
+def get_heat_comment(heat: float) -> str:
+    if heat < 0:
+        return "群成冰块啦，群主快开暖气"
+    elif heat < 10:
+        return "是冬天到了吗，好冷www"
+    elif heat < 20:
+        return "温度非常舒适，大家继续努力~"
+    elif heat < 30:
+        return "群热度达到最佳状态！"
+    elif heat < 39:
+        return "好热，群主快开空调！"
+    else:
+        return "请发送高温补贴喵。。"

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat.egg-info/PKG-INFO

@@ -0,0 +1,62 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-group-heat
+Version: 0.1.0
+Summary: 群热度统计插件 - 实时查询群内热度，生成昨日热度折线图
+Author-email: Wojusensei <3442006415@qq.com>
+License: MIT
+Project-URL: Homepage, https://github.com/Wojusensei/nonebot-plugin-group-heat
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: nonebot2>=2.3.0
+Requires-Dist: nonebot-adapter-onebot>=2.0.0
+Requires-Dist: nonebot-plugin-localstore>=0.5.0
+Requires-Dist: matplotlib>=3.5.0
+Requires-Dist: numpy>=1.21.0
+
+# nonebot-plugin-group-heat
+
+群热度统计插件 for NoneBot2
+
+## 功能
+
+- `/群热度` - 获取过去30分钟的群热度值，附带 bot 评价
+- `/昨日热度图` - 生成昨日每30分钟的热度折线图，并显示平均热度
+
+## 热度计算规则
+
+| 消息类型 | 热度值 |
+|---------|--------|
+| 文本消息 | +0.05 |
+| 表情包 | +0.20 |
+| 文件（图片/视频/语音等） | +0.30 |
+| 其他消息 | 不计入 |
+
+- 每个统计周期的初始热度为 -10°
+
+## 评价机制
+
+| 热度范围 | 评价 |
+|---------|------|
+| < 0 | 群成冰块啦，群主快开暖气 |
+| 0 ~ 10 | 是冬天到了吗，好冷www |
+| 10 ~ 20 | 温度非常舒适，大家继续努力~ |
+| 20 ~ 30 | 群热度达到最佳状态！ |
+| 30 ~ 39 | 好热，群主快开空调！ |
+| ≥ 39 | 请发送高温补贴喵。。 |
+
+## 安装
+
+```bash
+nb plugin install nonebot-plugin-group-heat
+pip install nonebot-plugin-group-heat(也可以)
+
+## 使用
+在群聊中发送以下命令：
+
+/群热度 - 查询过去30分钟的群热度
+
+/昨日热度图 - 获取昨日热度折线图
+
+## 开源协议
+MIT

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+pyproject.toml
+src/nonebot_plugin_group_heat/__init__.py
+src/nonebot_plugin_group_heat/config.py
+src/nonebot_plugin_group_heat/database.py
+src/nonebot_plugin_group_heat/heat_image.py
+src/nonebot_plugin_group_heat.egg-info/PKG-INFO
+src/nonebot_plugin_group_heat.egg-info/SOURCES.txt
+src/nonebot_plugin_group_heat.egg-info/dependency_links.txt
+src/nonebot_plugin_group_heat.egg-info/requires.txt
+src/nonebot_plugin_group_heat.egg-info/top_level.txt

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat.egg-info/requires.txt

@@ -0,0 +1,5 @@
+nonebot2>=2.3.0
+nonebot-adapter-onebot>=2.0.0
+nonebot-plugin-localstore>=0.5.0
+matplotlib>=3.5.0
+numpy>=1.21.0

nonebot_plugin_group_heat-0.1.0/src/nonebot_plugin_group_heat.egg-info/top_level.txt

@@ -0,0 +1 @@
+nonebot_plugin_group_heat