From 080534d03b26f4c71f828524aa03dcb70519a206 Mon Sep 17 00:00:00 2001 From: fujie Date: Wed, 11 Feb 2026 11:57:07 +0800 Subject: [PATCH] feat: implement Gist-based history tracking and enhanced stats categorization --- .github/workflows/community-stats.yml | 8 ++- scripts/openwebui_stats.py | 83 ++++++++++++++++++++++----- 2 files changed, 77 insertions(+), 14 deletions(-) diff --git a/.github/workflows/community-stats.yml b/.github/workflows/community-stats.yml index 8654b99..034b1ff 100644 --- a/.github/workflows/community-stats.yml +++ b/.github/workflows/community-stats.yml @@ -8,9 +8,13 @@ name: Community Stats on: - # 每小时整点运行 + # 定时任务 schedule: - cron: '0 * * * *' + # 推送时触发（用于测试任务分支） + push: + branches: + - feat/stats-history-and-refactor # 手动触发 workflow_dispatch: @@ -56,6 +60,8 @@ jobs: env: OPENWEBUI_API_KEY: ${{ secrets.OPENWEBUI_API_KEY }} OPENWEBUI_USER_ID: ${{ secrets.OPENWEBUI_USER_ID }} + GIST_TOKEN: ${{ secrets.GIST_TOKEN }} + GIST_ID: ${{ secrets.GIST_ID }} run: | python scripts/openwebui_stats.py diff --git a/scripts/openwebui_stats.py b/scripts/openwebui_stats.py index 9e055cb..e021592 100644 --- a/scripts/openwebui_stats.py +++ b/scripts/openwebui_stats.py @@ -47,16 +47,28 @@ class OpenWebUIStats: BASE_URL = "https://api.openwebui.com/api/v1" - def __init__(self, api_key: str, user_id: Optional[str] = None): + def __init__( + self, + api_key: str, + user_id: Optional[str] = None, + gist_token: Optional[str] = None, + gist_id: Optional[str] = None, + ): """ 初始化统计工具 Args: api_key: OpenWebUI API Key (JWT Token) user_id: 用户 ID，如果为 None 则从 token 中解析 + gist_token: GitHub Personal Access Token (用于读写 Gist) + gist_id: GitHub Gist ID """ self.api_key = api_key self.user_id = user_id or self._parse_user_id_from_token(api_key) + self.gist_token = gist_token + self.gist_id = gist_id + self.history_filename = "community-stats-history.json" + self.session = requests.Session() self.session.headers.update( { @@ -79,17 +91,34 @@ class OpenWebUIStats: ] def load_history(self) -> list: - """从文件加载历史记录""" + """加载历史记录 (优先尝试 Gist, 其次本地文件)""" + # 尝试从 Gist 加载 + if self.gist_token and self.gist_id: + try: + url = f"https://api.github.com/gists/{self.gist_id}" + headers = {"Authorization": f"token {self.gist_token}"} + resp = requests.get(url, headers=headers) + if resp.status_code == 200: + gist_data = resp.json() + file_info = gist_data.get("files", {}).get(self.history_filename) + if file_info: + content = file_info.get("content") + print(f"✅ 已从 Gist 加载历史记录 ({self.gist_id})") + return json.loads(content) + except Exception as e: + print(f"⚠️ 无法从 Gist 加载历史: {e}") + + # 降级：从本地加载 if self.history_file.exists(): try: with open(self.history_file, "r", encoding="utf-8") as f: return json.load(f) except Exception as e: - print(f"⚠️ 无法加载历史记录: {e}") + print(f"⚠️ 无法加载本地历史记录: {e}") return [] def save_history(self, stats: dict): - """保存当前快照到历史记录""" + """保存当前快照到历史记录 (优先保存到 Gist, 其次本地)""" history = self.load_history() today = get_beijing_time().strftime("%Y-%m-%d") @@ -104,31 +133,53 @@ class OpenWebUIStats: "points": stats.get("user", {}).get("total_points", 0), } - # 如果今天已存在，则更新；否则追加 + # 更新或追加数据点 + updated = False for i, item in enumerate(history): if item.get("date") == today: history[i] = snapshot + updated = True break - else: + if not updated: history.append(snapshot) - # 只保留最近 90 天的历史 + # 限制长度 (90天) history = history[-90:] + # 尝试保存到 Gist + if self.gist_token and self.gist_id: + try: + url = f"https://api.github.com/gists/{self.gist_id}" + headers = {"Authorization": f"token {self.gist_token}"} + payload = { + "files": { + self.history_filename: { + "content": json.dumps(history, ensure_ascii=False, indent=2) + } + } + } + resp = requests.patch(url, headers=headers, json=payload) + if resp.status_code == 200: + print(f"✅ 历史记录已同步至 Gist ({self.gist_id})") + return + except Exception as e: + print(f"⚠️ 同步至 Gist 失败: {e}") + + # 降级：保存到本地 with open(self.history_file, "w", encoding="utf-8") as f: json.dump(history, f, ensure_ascii=False, indent=2) - print(f"✅ 历史快照已更新 ({today})") + print(f"✅ 历史记录已更新至本地 ({today})") def get_stat_delta(self, stats: dict) -> dict: - """计算相对于上次记录的增长""" + """计算相对于上次记录的增长 (24h)""" history = self.load_history() - if len(history) < 2: + if not history: return {} - # 获取上一次的快照（倒数第二个，因为当前可能已经存入倒数第一个） - # 或者如果还没存入，就是倒数第一个 today = get_beijing_time().strftime("%Y-%m-%d") prev = None + + # 查找非今天的最后一笔数据作为基准 for item in reversed(history): if item.get("date") != today: prev = item @@ -754,9 +805,15 @@ def main(): print(" 例如: b15d1348-4347-42b4-b815-e053342d6cb0") return 1 + # 获取 Gist 配置 (用于存储历史记录) + gist_token = os.getenv("GIST_TOKEN") + gist_id = os.getenv("GIST_ID") + # 初始化 - stats_client = OpenWebUIStats(api_key, user_id) + stats_client = OpenWebUIStats(api_key, user_id, gist_token, gist_id) print(f"🔍 用户 ID: {stats_client.user_id}") + if gist_id: + print(f"📦 Gist 存储已启用: {gist_id}") # 获取所有帖子 print("📥 正在获取帖子数据...")