feat: 公告和更新检查

pkg/utils/announce.py

@@ -0,0 +1,106 @@
+from __future__ import annotations
+
+import json
+import typing
+import os
+import base64
+
+import pydantic
+import requests
+
+from ..core import app
+
+
+class Announcement(pydantic.BaseModel):
+    """公告"""
+    
+    id: int
+
+    time: str
+
+    timestamp: int
+
+    content: str
+
+    enabled: typing.Optional[bool] = True
+
+    def to_dict(self) -> dict:
+        return {
+            "id": self.id,
+            "time": self.time,
+            "timestamp": self.timestamp,
+            "content": self.content,
+            "enabled": self.enabled
+        }
+
+
+class AnnouncementManager:
+    """公告管理器"""
+
+    ap: app.Application = None
+
+    def __init__(self, ap: app.Application):
+        self.ap = ap
+
+    async def fetch_all(
+        self
+    ) -> list[Announcement]:
+        """获取所有公告"""
+        resp = requests.get(
+            url="https://api.github.com/repos/RockChinQ/QChatGPT/contents/res/announcement.json",
+            proxies=self.ap.proxy_mgr.get_forward_proxies(),
+            timeout=5
+        )
+        obj_json = resp.json()
+        b64_content = obj_json["content"]
+        # 解码
+        content = base64.b64decode(b64_content).decode("utf-8")
+
+        return [Announcement(**item) for item in json.loads(content)]
+
+    async def fetch_saved(
+        self
+    ) -> list[Announcement]:
+        if not os.path.exists("res/announcement_saved.json"):
+            with open("res/announcement_saved.json", "w", encoding="utf-8") as f:
+                f.write("[]")
+
+        with open("res/announcement_saved.json", "r", encoding="utf-8") as f:
+            content = f.read()
+
+        if not content:
+            content = '[]'
+
+        return [Announcement(**item) for item in json.loads(content)]
+
+    async def write_saved(
+        self,
+        content: list[Announcement]
+    ):
+
+        with open("res/announcement_saved.json", "w", encoding="utf-8") as f:
+            f.write(json.dumps([
+                item.to_dict() for item in content
+            ], indent=4, ensure_ascii=False))
+
+    async def fetch_new(
+        self
+    ) -> list[Announcement]:
+        """获取新公告"""
+        all = await self.fetch_all()
+        saved = await self.fetch_saved()
+
+        to_show: list[Announcement] = []
+
+        for item in all:
+            # 遍历saved检查是否有相同id的公告
+            for saved_item in saved:
+                if saved_item.id == item.id:
+                    break
+            else:
+                if item.enabled:
+                    # 没有相同id的公告
+                    to_show.append(item)
+
+        await self.write_saved(all)
+        return to_show

pkg/utils/announcement.py

@@ -1,68 +0,0 @@
-import base64
-import os
-import json
-
-import requests
-
-
-def read_latest() -> list:
-    import pkg.utils.network as network
-    resp = requests.get(
-        url="https://api.github.com/repos/RockChinQ/QChatGPT/contents/res/announcement.json",
-        proxies=network.wrapper_proxies()
-    )
-    obj_json = resp.json()
-    b64_content = obj_json["content"]
-    # 解码
-    content = base64.b64decode(b64_content).decode("utf-8")
-    return json.loads(content)
-
-
-def read_saved() -> list:
-    # 已保存的在res/announcement_saved
-    # 检查是否存在
-    if not os.path.exists("res/announcement_saved.json"):
-        with open("res/announcement_saved.json", "w", encoding="utf-8") as f:
-            f.write("[]")
-
-    with open("res/announcement_saved.json", "r", encoding="utf-8") as f:
-        content = f.read()
-
-    return json.loads(content)
-
-
-def write_saved(content: list):
-    # 已保存的在res/announcement_saved
-    with open("res/announcement_saved.json", "w", encoding="utf-8") as f:
-        f.write(json.dumps(content, indent=4, ensure_ascii=False))
-
-
-def fetch_new() -> list:
-    latest = read_latest()
-    saved = read_saved()
-
-    to_show: list = []
-
-    for item in latest:
-        # 遍历saved检查是否有相同id的公告
-        for saved_item in saved:
-            if saved_item["id"] == item["id"]:
-                break
-        else:
-            # 没有相同id的公告
-            to_show.append(item)
-
-    write_saved(latest)
-    return to_show
-
-
-if __name__ == '__main__':
-    
-    resp = requests.get(
-        url="https://api.github.com/repos/RockChinQ/QChatGPT/contents/res/announcement.json",
-    )
-    obj_json = resp.json()
-    b64_content = obj_json["content"]
-    # 解码
-    content = base64.b64decode(b64_content).decode("utf-8")
-    print(json.dumps(json.loads(content), indent=4, ensure_ascii=False))

pkg/utils/version.py

@@ -49,7 +49,8 @@ class VersionManager:
         """获取发行列表"""
         rls_list_resp = requests.get(
             url="https://api.github.com/repos/RockChinQ/QChatGPT/releases",
-            proxies=self.ap.proxy_mgr.get_forward_proxies()
+            proxies=self.ap.proxy_mgr.get_forward_proxies(),
+            timeout=5
         )
 
         rls_list = rls_list_resp.json()