diff --git a/src/plugins/built_in/maizone_refactored/plugin.py b/src/plugins/built_in/maizone_refactored/plugin.py index ee0826f9c..7513fff11 100644 --- a/src/plugins/built_in/maizone_refactored/plugin.py +++ b/src/plugins/built_in/maizone_refactored/plugin.py @@ -43,7 +43,6 @@ class MaiZoneRefactoredPlugin(BasePlugin): "plugin": {"enable": ConfigField(type=bool, default=True, description="是否启用插件")}, "models": { "text_model": ConfigField(type=str, default="maizone", description="生成文本的模型名称"), - "vision_model": ConfigField(type=str, default="sf-glm-4.5v", description="识别图片的模型名称（建议使用model_config.toml中配置的视觉模型）"), "siliconflow_apikey": ConfigField(type=str, default="", description="硅基流动AI生图API密钥"), }, "send": { diff --git a/src/plugins/built_in/maizone_refactored/services/content_service.py b/src/plugins/built_in/maizone_refactored/services/content_service.py index f4793477b..b446d1dbb 100644 --- a/src/plugins/built_in/maizone_refactored/services/content_service.py +++ b/src/plugins/built_in/maizone_refactored/services/content_service.py @@ -245,14 +245,15 @@ class ContentService: image_format = kind.extension image_base64 = base64.b64encode(image_bytes).decode("utf-8") - vision_model_name = self.get_config("models.vision_model", "vlm") + # 优先从全局配置读取视觉模型，如果未配置，则使用默认的 "vlm" + vision_model_name = config_api.get_global_config("model.vision.default_model", "vlm") - # 使用 llm_api 获取模型配置，支持自动fallback到备选模型 + # 使用 llm_api 获取模型配置 models = llm_api.get_available_models() vision_model_config = models.get(vision_model_name) if not vision_model_config: - logger.error(f"未找到视觉模型配置: {vision_model_name}") + logger.error(f"未在 model_config.toml 中找到视觉模型配置: {vision_model_name}") return None vision_model_config.temperature = 0.3 diff --git a/src/plugins/built_in/maizone_refactored/services/qzone_service.py b/src/plugins/built_in/maizone_refactored/services/qzone_service.py index 13e022866..476be1129 100644 --- a/src/plugins/built_in/maizone_refactored/services/qzone_service.py +++ b/src/plugins/built_in/maizone_refactored/services/qzone_service.py @@ -48,7 +48,7 @@ class QZoneService: content_service: ContentService, image_service: ImageService, cookie_service: CookieService, - reply_tracker: ReplyTrackerService = None, + reply_tracker: ReplyTrackerService | None = None, ): self.get_config = get_config self.content_service = content_service @@ -128,7 +128,8 @@ class QZoneService: target_person_id = await person_api.get_person_id_by_name(target_name) if not target_person_id: return {"success": False, "message": f"找不到名为'{target_name}'的好友"} - target_qq = await person_api.get_person_value(target_person_id, "user_id") + person_info = await person_api.get_person_info(target_person_id) + target_qq = person_info.get("user_id") if not target_qq: return {"success": False, "message": f"好友'{target_name}'没有关联QQ号"} @@ -155,7 +156,7 @@ class QZoneService: total_liked = 0 total_commented = 0 for feed in feeds: - result = await self._process_single_feed(feed, api_client, target_qq, target_name) + result = await self._process_single_feed(feed, api_client, str(target_qq), target_name) if result["liked"]: total_liked += 1 if result["commented"]: @@ -254,7 +255,7 @@ class QZoneService: if not target_qq or str(target_qq) == str(qq_account): # 确保不重复处理自己的 continue - result = await self._process_single_feed(feed, api_client, target_qq, target_qq) + result = await self._process_single_feed(feed, api_client, str(target_qq), str(target_qq)) monitor_stats["total"] += 1 if result.get("liked"): monitor_stats["liked"] += 1 @@ -1151,28 +1152,28 @@ class QZoneService: like_btn = soup.find("a", class_="qz_like_btn_v3") is_liked = False - if like_btn and isinstance(like_btn, bs4.Tag): - is_liked = like_btn.get("data-islike") == "1" + if isinstance(like_btn, bs4.Tag) and like_btn.get("data-islike") == "1": + is_liked = True if is_liked: continue text_div = soup.find("div", class_="f-info") - text = text_div.get_text(strip=True) if text_div else "" + text = text_div.get_text(strip=True) if isinstance(text_div, bs4.Tag) else "" # --- 借鉴原版插件的精确图片提取逻辑 --- image_urls = [] img_box = soup.find("div", class_="img-box") - if img_box: + if isinstance(img_box, bs4.Tag): for img in img_box.find_all("img"): - src = img.get("src") - # 排除QQ空间的小图标和表情 - if src and "qzonestyle.gtimg.cn" not in src: - image_urls.append(src) + if isinstance(img, bs4.Tag): + src = img.get("src") + if src and isinstance(src, str) and "qzonestyle.gtimg.cn" not in src: + image_urls.append(src) # 视频封面也视为图片 video_thumb = soup.select_one("div.video-img img") - if video_thumb and "src" in video_thumb.attrs: + if isinstance(video_thumb, bs4.Tag) and "src" in video_thumb.attrs: image_urls.append(video_thumb["src"]) # 去重 @@ -1181,11 +1182,13 @@ class QZoneService: comments = [] comment_divs = soup.find_all("div", class_="f-single-comment") for comment_div in comment_divs: + if not isinstance(comment_div, bs4.Tag): + continue # --- 处理主评论 --- author_a = comment_div.find("a", class_="f-nick") content_span = comment_div.find("span", class_="f-re-con") - if author_a and content_span: + if isinstance(author_a, bs4.Tag) and isinstance(content_span, bs4.Tag): comments.append( { "qq_account": str(comment_div.get("data-uin", "")), @@ -1199,21 +1202,23 @@ class QZoneService: # --- 处理这条主评论下的所有回复 --- reply_divs = comment_div.find_all("div", class_="f-single-re") for reply_div in reply_divs: + if not isinstance(reply_div, bs4.Tag): + continue reply_author_a = reply_div.find("a", class_="f-nick") reply_content_span = reply_div.find("span", class_="f-re-con") - if reply_author_a and reply_content_span: + if isinstance(reply_author_a, bs4.Tag) and isinstance(reply_content_span, bs4.Tag): comments.append( { "qq_account": str(reply_div.get("data-uin", "")), "nickname": reply_author_a.get_text(strip=True), "content": reply_content_span.get_text(strip=True).lstrip( ": " - ), # 移除回复内容前多余的冒号和空格 + ), "comment_tid": reply_div.get("data-tid", ""), "parent_tid": reply_div.get( "data-parent-tid", comment_div.get("data-tid", "") - ), # 如果没有父ID，则将父ID设为主评论ID + ), } ) diff --git a/src/plugins/built_in/maizone_refactored/services/reply_tracker_service.py b/src/plugins/built_in/maizone_refactored/services/reply_tracker_service.py index 8fa0ce1e4..0c6e9ef22 100644 --- a/src/plugins/built_in/maizone_refactored/services/reply_tracker_service.py +++ b/src/plugins/built_in/maizone_refactored/services/reply_tracker_service.py @@ -65,9 +65,9 @@ class ReplyTrackerService: try: if self.reply_record_file.exists(): try: - with open(self.reply_record_file, encoding="utf-8") as f: - file_content = f.read().strip() - if not file_content: # 文件为空 + with open(self.reply_record_file, "rb") as f: + file_content = f.read() + if not file_content.strip(): # 文件为空 logger.warning("回复记录文件为空，将创建新的记录") self.replied_comments = {} return @@ -118,8 +118,8 @@ class ReplyTrackerService: temp_file = self.reply_record_file.with_suffix(".tmp") # 先写入临时文件 - with open(temp_file, "w", encoding="utf-8"): - orjson.dumps(self.replied_comments, option=orjson.OPT_INDENT_2 | orjson.OPT_NON_STR_KEYS).decode("utf-8") + with open(temp_file, "wb") as f: + f.write(orjson.dumps(self.replied_comments, option=orjson.OPT_INDENT_2 | orjson.OPT_NON_STR_KEYS)) # 如果写入成功，重命名为正式文件 if temp_file.stat().st_size > 0: # 确保写入成功