From c071088952dfb36a0d32f7032300277731a9c052 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Tue, 23 Apr 2024 10:52:59 +0800
Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E4=BF=AE=E5=A4=8D=E9=A6=92?=
 =?UTF-8?q?=E5=A4=B4=E6=B6=88=E6=81=AF=E7=BB=9F=E8=AE=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 package.json                                   | 3 ++-
 plugins/sitestatistic/__init__.py              | 2 +-
 plugins/sitestatistic/siteuserinfo/__init__.py | 3 +++
 plugins/sitestatistic/siteuserinfo/mtorrent.py | 3 ++-
 4 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/package.json b/package.json
index cbcbdad..8fa1438 100644
--- a/package.json
+++ b/package.json
@@ -21,11 +21,12 @@
     "SiteStatistic": {
         "name": "站点数据统计",
         "description": "自动统计和展示站点数据。",
-        "version": "2.7",
+        "version": "2.8",
         "icon": "statistic.png",
         "author": "lightolly",
         "level": 2,
         "history": {
+            "v2.8": "修复馒头未读消息统计",
             "v2.7": "修复憨憨种子信息只统计第一页的问题，增加移除失效统计选项",
             "v2.6": "支持馒头新架构数据统计"
         }
diff --git a/plugins/sitestatistic/__init__.py b/plugins/sitestatistic/__init__.py
index 1daf5cc..b07c5d8 100644
--- a/plugins/sitestatistic/__init__.py
+++ b/plugins/sitestatistic/__init__.py
@@ -43,7 +43,7 @@ class SiteStatistic(_PluginBase):
     # 插件图标
     plugin_icon = "statistic.png"
     # 插件版本
-    plugin_version = "2.7"
+    plugin_version = "2.8"
     # 插件作者
     plugin_author = "lightolly"
     # 作者主页
diff --git a/plugins/sitestatistic/siteuserinfo/__init__.py b/plugins/sitestatistic/siteuserinfo/__init__.py
index 47548da..e98a1e2 100644
--- a/plugins/sitestatistic/siteuserinfo/__init__.py
+++ b/plugins/sitestatistic/siteuserinfo/__init__.py
@@ -217,6 +217,9 @@ class ISiteUserInfo(metaclass=ABCMeta):
                         msg_links
                     )
                 unread_msg_links.extend(msg_links)
+        # 重新更新未读消息数（99999表示有消息但数量未知）
+        if self.message_unread == 99999:
+            self.message_unread = len(unread_msg_links)
         # 解析未读消息内容
         for msg_link in unread_msg_links:
             logger.debug(f"{self.site_name} 信息链接 {msg_link}")
diff --git a/plugins/sitestatistic/siteuserinfo/mtorrent.py b/plugins/sitestatistic/siteuserinfo/mtorrent.py
index 758fc7c..4881418 100644
--- a/plugins/sitestatistic/siteuserinfo/mtorrent.py
+++ b/plugins/sitestatistic/siteuserinfo/mtorrent.py
@@ -102,7 +102,8 @@ class MTorrentSiteUserInfo(ISiteUserInfo):
         self.download = int(user_info.get("memberCount", {}).get("downloaded") or '0')
         self.ratio = user_info.get("memberCount", {}).get("shareRate") or 0
         self.bonus = user_info.get("memberCount", {}).get("bonus") or 0
-        self.message_unread = 1
+        # 需要解析消息，但不确定消息条数
+        self.message_unread = 99999
 
         self._torrent_seeding_params = {
             "pageNumber": 1,