功能：通过 HTTP API 实现 Danding_QqPush 插件，用于 QQ 群通知

- 增加了通过外部 HTTP API 向 QQ 群组发送消息的核心功能。 - 实现了对长文本消息的图片渲染，以避免被认定为垃圾信息。 - 支持在消息中提及特定的 QQ 用户。 - 创建了用于 API 令牌和图片渲染设置的配置选项。 - 开发了一个测试脚本以验证 API 功能。 - 对现有代码进行了重构，以提高组织性和可维护性。
2026-01-20 21:19:05 +08:00
parent 4a944316fe
commit dedc872f1b
14 changed files with 1065 additions and 48 deletions
--- a/danding_bot/plugins/danding_qqpush/image_render.py
+++ b/danding_bot/plugins/danding_qqpush/image_render.py
@@ -0,0 +1,207 @@
+"""图片生成模块 - 将文本渲染为图片"""
+from PIL import Image, ImageDraw, ImageFont
+import io
+import base64
+from typing import Tuple
+
+
+class ImageRenderer:
+    """图片渲染器"""
+    
+    def __init__(
+        self,
+        width: int = 800,
+        font_size: int = 24,
+        padding: int = 30,
+        line_spacing: float = 1.4,
+        bg_color: Tuple[int, int, int] = (252, 252, 252),
+        text_color: Tuple[int, int, int] = (0, 0, 0),
+        font_paths: list = None
+    ):
+        """
+        初始化图片渲染器
+        
+        Args:
+            width: 图片宽度
+            font_size: 字体大小
+            padding: 内边距
+            line_spacing: 行距倍数
+            bg_color: 背景颜色 (R, G, B)
+            text_color: 文本颜色 (R, G, B)
+            font_paths: 字体文件路径列表
+        """
+        self.width = width
+        self.font_size = font_size
+        self.padding = padding
+        self.line_spacing = line_spacing
+        self.bg_color = bg_color
+        self.text_color = text_color
+        self.font_paths = font_paths or []
+        
+        # 加载字体
+        self.font = self._load_font()
+    
+    def _load_font(self) -> ImageFont.FreeTypeFont:
+        """加载字体"""
+        return self._load_font_by_size(self.font_size)
+    
+    def _load_font_by_size(self, font_size: int) -> ImageFont.FreeTypeFont:
+        """
+        加载指定大小的字体
+        
+        Args:
+            font_size: 字体大小
+            
+        Returns:
+            字体对象
+        """
+        for font_path in self.font_paths:
+            try:
+                font = ImageFont.truetype(font_path, font_size)
+                return font
+            except Exception:
+                continue
+        
+        # 如果都失败了，使用默认字体
+        return ImageFont.load_default()
+    
+    def _calculate_text_dimensions(self, text: str) -> Tuple[int, int]:
+        """
+        计算文本的尺寸
+        
+        Args:
+            text: 文本内容
+            
+        Returns:
+            (宽度, 高度)
+        """
+        # 创建一个临时图片用于计算
+        test_img = Image.new('RGB', (1, 1), color=self.bg_color)
+        test_draw = ImageDraw.Draw(test_img)
+        
+        bbox = test_draw.textbbox((0, 0), text, font=self.font)
+        width = bbox[2] - bbox[0]
+        height = bbox[3] - bbox[1]
+        
+        return width, height
+    
+    def _smart_text_wrap(self, text: str, max_width: int) -> list:
+        """
+        智能文本换行
+        
+        Args:
+            text: 文本内容
+            max_width: 最大宽度
+            
+        Returns:
+            换行后的文本列表
+        """
+        lines = []
+        current_line = ""
+        current_width = 0
+        
+        for char in text:
+            char_width, _ = self._calculate_text_dimensions(char)
+            
+            if current_width + char_width <= max_width:
+                current_line += char
+                current_width += char_width
+            else:
+                if current_line:
+                    lines.append(current_line)
+                current_line = char
+                current_width = char_width
+        
+        if current_line:
+            lines.append(current_line)
+        
+        return lines
+    
+    def render(self, text: str) -> bytes:
+        """
+        将文本渲染为图片
+        
+        Args:
+            text: 文本内容（已处理换行符）
+            
+        Returns:
+            图片的字节数据
+        """
+        if not text:
+            text = "（空消息）"
+        
+        # 计算有效宽度
+        effective_width = self.width - (2 * self.padding)
+        
+        # 按段落处理
+        all_lines = []
+        for paragraph in text.split('\n'):
+            if not paragraph:
+                all_lines.append("")
+                continue
+            
+            wrapped_lines = self._smart_text_wrap(paragraph, effective_width)
+            all_lines.extend(wrapped_lines)
+        
+        # 计算行高
+        _, line_height = self._calculate_text_dimensions("测试")
+        total_line_height = int(line_height * self.line_spacing)
+        
+        # 标题相关
+        title = "蛋定助手通知您："
+        title_font_size = int(self.font_size * 1.3)  # 标题字体放大1.3倍
+        title_font = self._load_font_by_size(title_font_size)
+        title_height = int(line_height * 1.5)  # 标题区域高度
+        divider_height = 2  # 分割线高度
+        divider_spacing = 10  # 分割线与标题的间距
+        
+        # 计算总高度（包含标题区域）
+        content_height = (len(all_lines) * total_line_height) + (2 * self.padding)
+        header_height = title_height + divider_spacing + divider_height + self.padding
+        total_height = content_height + header_height
+        height = max(total_height, 200)  # 最小高度
+        
+        # 创建图片
+        image = Image.new('RGB', (self.width, height), color=self.bg_color)
+        draw = ImageDraw.Draw(image)
+        
+        # 绘制标题（加粗效果通过增大字体实现）
+        title_x = self.padding
+        title_y = self.padding
+        draw.text((title_x, title_y), title, font=title_font, fill=self.text_color)
+        
+        # 绘制分割线
+        divider_y = title_y + title_height + 10
+        draw.line(
+            [(self.padding, divider_y), (self.width - self.padding, divider_y)],
+            fill=(200, 200, 200),
+            width=divider_height
+        )
+        
+        # 绘制文本内容
+        y = divider_y + divider_spacing + self.padding
+        for line in all_lines:
+            if line:  # 跳过空行
+                draw.text((self.padding, y), line, font=self.font, fill=self.text_color)
+            y += total_line_height
+        
+        # 转换为字节流
+        img_byte_arr = io.BytesIO()
+        image.save(img_byte_arr, format='PNG', quality=95)
+        img_byte_arr.seek(0)
+        
+        return img_byte_arr.getvalue()
+    
+    def render_to_base64(self, text: str) -> str:
+        """
+        将文本渲染为图片并返回 base64 编码
+        
+        Args:
+            text: 文本内容
+            
+        Returns:
+            base64 编码的图片数据
+        """
+        img_bytes = self.render(text)
+        base64_str = base64.b64encode(img_bytes).decode('utf-8')
+        return f"base64://{base64_str}"