Upload 3 files

app.py

@@ -55,20 +55,22 @@ total_tokens = {
     "total": 0         # 总token统计
 }
 
-# 计算点信息
-compute_points = {
-    "left": 0,          # 剩余计算点
-    "total": 0,         # 总计算点
-    "used": 0,          # 已使用计算点
-    "percentage": 0,    # 使用百分比
-    "last_update": None # 最后更新时间
-}
-
-# 计算点使用日志
-compute_points_log = {
-    "columns": {},      # 列名
-    "log": []           # 日志数据
-}
+# 计算点信息 (现在是列表)
+compute_points = []
+# {
+#     "left": 0,          # 剩余计算点
+#     "total": 0,         # 总计算点
+#     "used": 0,          # 已使用计算点
+#     "percentage": 0,    # 使用百分比
+#     "last_update": None # 最后更新时间
+# }
+
+# 计算点使用日志 (现在是列表)
+compute_points_log = []
+# {
+#     "columns": {},      # 列名
+#     "log": []           # 日志数据
+# }
 
 
 # 记录启动时间
@@ -821,105 +823,68 @@ def update_model_stats(model, prompt_tokens, completion_tokens):
 
 # 获取计算点信息
 def get_compute_points():
-    global compute_points, USER_DATA
+    global compute_points, compute_points_log
     
-    if USER_NUM == 0:
-        return
+    # 限制只获取前两个用户的数据
+    users_to_fetch = USER_DATA[:2]
     
-    try:
-        # 使用第一个用户的会话和凭据
-        session, cookies, session_token, _, _ = USER_DATA[0]
-        
-        # 检查token是否有效
-        if is_token_expired(session_token):
-            session_token = refresh_token(session, cookies)
-            if not session_token:
-                print("刷新token失败，无法获取计算点信息")
-                return
-            USER_DATA[0] = (session, cookies, session_token, USER_DATA[0][3], USER_DATA[0][4])
-        
-        headers = {
-            "accept": "application/json, text/plain, */*",
-            "accept-language": "zh-CN,zh;q=0.9",
-            "baggage": f"sentry-environment=production,sentry-release=93da8385541a6ce339b1f41b0c94428c70657e22,sentry-public_key=3476ea6df1585dd10e92cdae3a66ff49,sentry-trace_id={TRACE_ID}",
-            "reai-ui": "1",
-            "sec-ch-ua": "\"Chromium\";v=\"116\", \"Not)A;Brand\";v=\"24\", \"Google Chrome\";v=\"116\"",
-            "sec-ch-ua-mobile": "?0",
-            "sec-ch-ua-platform": "\"Windows\"",
-            "sec-fetch-dest": "empty",
-            "sec-fetch-mode": "cors",
-            "sec-fetch-site": "same-origin",
-            "sentry-trace": SENTRY_TRACE,
-            "session-token": session_token,
-            "x-abacus-org-host": "apps",
-            "cookie": cookies
-        }
-        
-        response = session.get(
-            COMPUTE_POINTS_URL,
-            headers=headers
-        )
-        
-        if response.status_code == 200:
-            result = response.json()
-            if result.get("success") and "result" in result:
-                data = result["result"]
-                compute_points["left"] = data.get("computePointsLeft", 0)
-                compute_points["total"] = data.get("totalComputePoints", 0)
-                compute_points["used"] = compute_points["total"] - compute_points["left"]
-                compute_points["percentage"] = round((compute_points["used"] / compute_points["total"]) * 100, 2) if compute_points["total"] > 0 else 0
-                compute_points["last_update"] = datetime.now()
-                print(f"计算点信息更新成功: 剩余 {compute_points['left']}, 总计 {compute_points['total']}")
-                
-                # 获取计算点使用日志
-                get_compute_points_log(session, cookies, session_token)
-            else:
-                print(f"获取计算点信息失败: {result.get('error', '未知错误')}")
-        else:
-            print(f"获取计算点信息失败，状态码: {response.status_code}")
-    except Exception as e:
-        print(f"获取计算点信息异常: {e}")
+    new_compute_points = []
+    new_compute_points_log = []
 
-# 获取计算点使用日志
-def get_compute_points_log(session, cookies, session_token):
-    global compute_points_log
-    
-    try:
-        headers = {
-            "accept": "application/json, text/plain, */*",
-            "accept-language": "zh-CN,zh;q=0.9",
-            "content-type": "application/json",
-            "reai-ui": "1",
-            "sec-ch-ua": "\"Chromium\";v=\"116\", \"Not)A;Brand\";v=\"24\", \"Google Chrome\";v=\"116\"",
-            "sec-ch-ua-mobile": "?0",
-            "sec-ch-ua-platform": "\"Windows\"",
-            "sec-fetch-dest": "empty",
-            "sec-fetch-mode": "cors",
-            "sec-fetch-site": "same-site",
-            "session-token": session_token,
-            "x-abacus-org-host": "apps",
-            "cookie": cookies
+    for user_index, user_config in enumerate(users_to_fetch):
+        user_compute_points = {
+            "left": 0, "total": 0, "used": 0, "percentage": 0, "last_update": None, "error": None
         }
-        
-        response = session.post(
-            COMPUTE_POINTS_LOG_URL,
-            headers=headers,
-            json={"byLlm": True}
-        )
-        
-        if response.status_code == 200:
-            result = response.json()
-            if result.get("success") and "result" in result:
-                data = result["result"]
-                compute_points_log["columns"] = data.get("columns", {})
-                compute_points_log["log"] = data.get("log", [])
-                print(f"计算点使用日志更新成功，获取到 {len(compute_points_log['log'])} 条记录")
-            else:
-                print(f"获取计算点使用日志失败: {result.get('error', '未知错误')}")
-        else:
-            print(f"获取计算点使用日志失败，状态码: {response.status_code}")
-    except Exception as e:
-        print(f"获取计算点使用日志异常: {e}")
+        user_compute_points_log = {
+            "columns": {}, "log": [], "error": None
+        }
+
+        try:
+            headers = {
+                "Cookie": user_config["cookies"],
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
+            }
+            
+            # 获取计算点信息
+            compute_url = "https://abacus.art/api/trpc/user.getComputePoints"
+            response = requests.get(compute_url, headers=headers)
+            response.raise_for_status()
+            data = response.json()
+            
+            points_data = data.get("result", {}).get("data", {})
+            user_compute_points["left"] = points_data.get("left", 0)
+            user_compute_points["total"] = points_data.get("total", 0)
+            user_compute_points["used"] = points_data.get("used", 0)
+            user_compute_points["percentage"] = points_data.get("percentage", 0)
+            user_compute_points["last_update"] = datetime.now()
+
+            # 获取计算点使用日志
+            log_url = "https://abacus.art/api/trpc/user.getComputePointsLog?batch=1&input=%7B%220%22%3A%7B%22json%22%3Anull%2C%22meta%22%3A%7B%22values%22%3A%5B%22undefined%22%5D%7D%7D%7D"
+            response = requests.get(log_url, headers=headers)
+            response.raise_for_status()
+            log_data = response.json()
+            
+            log_result = log_data[0].get("result", {}).get("data", {}).get("json", {})
+            user_compute_points_log["columns"] = log_result.get("columns", {})
+            user_compute_points_log["log"] = log_result.get("log", [])
+
+        except requests.exceptions.RequestException as e:
+            error_message = f"用户 {user_index + 1} 获取计算点信息异常: {e}"
+            print(error_message)
+            user_compute_points["error"] = str(e)
+            user_compute_points_log["error"] = str(e)
+        except Exception as e:
+            error_message = f"用户 {user_index + 1} 处理计算点信息时发生未知错误: {e}"
+            print(error_message)
+            user_compute_points["error"] = str(e)
+            user_compute_points_log["error"] = str(e)
+            
+        new_compute_points.append(user_compute_points)
+        new_compute_points_log.append(user_compute_points_log)
+
+    # 更新全局变量
+    compute_points = new_compute_points
+    compute_points_log = new_compute_points_log
 
 
 # 添加登录相关路由

templates/dashboard.html

@@ -550,43 +550,67 @@
         </div>
         
         <div class="grid">
+            {% if compute_points %}
+                {% for user_points in compute_points %}
             <div class="card">
                 <div class="card-header">
                     <h2 class="card-title">
                         <span class="card-icon">💰</span>
-                        计算点信息
+                        用户 {{ loop.index }} 计算点信息
                     </h2>
                 </div>
+                {% if user_points.error %}
+                <div class="status-item">
+                    <span class="status-label">错误</span>
+                    <span class="status-value danger">获取失败: {{ user_points.error }}</span>
+                </div>
+                {% else %}
                 <div class="status-item">
                     <span class="status-label">总计算点</span>
-                    <span class="status-value compute-points">{{ compute_points.total|int }}</span>
+                    <span class="status-value compute-points">{{ user_points.total|int }}</span>
                 </div>
                 <div class="status-item">
                     <span class="status-label">已使用</span>
-                    <span class="status-value compute-points">{{ compute_points.used|int }}</span>
+                    <span class="status-value compute-points">{{ user_points.used|int }}</span>
                 </div>
                 <div class="status-item">
                     <span class="status-label">剩余</span>
-                    <span class="status-value compute-points">{{ compute_points.left|int }}</span>
+                    <span class="status-value compute-points">{{ user_points.left|int }}</span>
                 </div>
                 <div class="status-item">
                     <span class="status-label">使用比例</span>
                     <div style="width: 100%; text-align: right;">
-                        <span class="status-value compute-points {% if compute_points.percentage > 80 %}danger{% elif compute_points.percentage > 50 %}warning{% endif %}">
-                            {{ compute_points.percentage }}%
+                        <span class="status-value compute-points {% if user_points.percentage > 80 %}danger{% elif user_points.percentage > 50 %}warning{% endif %}">
+                            {{ user_points.percentage }}%
                         </span>
                         <div class="progress-container">
-                            <div class="progress-bar {% if compute_points.percentage > 80 %}danger{% elif compute_points.percentage > 50 %}warning{% endif %}" style="width: {{ compute_points.percentage }}%"></div>
+                            <div class="progress-bar {% if user_points.percentage > 80 %}danger{% elif user_points.percentage > 50 %}warning{% endif %}" style="width: {{ user_points.percentage }}%"></div>
                         </div>
                     </div>
                 </div>
-                {% if compute_points.last_update %}
+                {% if user_points.last_update %}
                 <div class="status-item">
                     <span class="status-label">最后更新时间</span>
-                    <span class="status-value">{{ compute_points.last_update.strftime('%Y-%m-%d %H:%M:%S') }}</span>
+                    <span class="status-value">{{ user_points.last_update.strftime('%Y-%m-%d %H:%M:%S') }}</span>
                 </div>
                 {% endif %}
+                {% endif %} {# End if user_points.error #}
+            </div>
+                {% endfor %}
+            {% else %}
+            <div class="card">
+                <div class="card-header">
+                    <h2 class="card-title">
+                        <span class="card-icon">💰</span>
+                        计算点信息
+                    </h2>
+                </div>
+                <div class="status-item">
+                    <span class="status-label">状态</span>
+                    <span class="status-value warning">暂无计算点数据或未配置用户</span>
+                </div>
             </div>
+            {% endif %} {# End if compute_points #}
             
             <div class="card">
                 <div class="card-header">