Upload 4 files

app.py

@@ -70,6 +70,8 @@ compute_points_log = {
     "log": []           # 日志数据
 }
 
+# 多用户计算点信息
+users_compute_points = []
 
 # 记录启动时间
 START_TIME = datetime.now()
@@ -821,64 +823,97 @@ def update_model_stats(model, prompt_tokens, completion_tokens):
 
 # 获取计算点信息
 def get_compute_points():
-    global compute_points, USER_DATA
+    global compute_points, USER_DATA, users_compute_points
     
     if USER_NUM == 0:
         return
     
-    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)
+    # 清空用户计算点列表
+    users_compute_points = []
+    
+    # 累计总计算点
+    total_left = 0
+    total_points = 0
+    
+    # 获取每个用户的计算点信息
+    for i, user_data in enumerate(USER_DATA):
+        try:
+            session, cookies, session_token, _, _ = user_data
+            
+            # 检查token是否有效
+            if is_token_expired(session_token):
+                session_token = refresh_token(session, cookies)
+                if not session_token:
+                    print(f"用户{i+1}刷新token失败，无法获取计算点信息")
+                    continue
+                USER_DATA[i] = (session, cookies, session_token, user_data[3], user_data[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"]
+                    left = data.get("computePointsLeft", 0)
+                    total = data.get("totalComputePoints", 0)
+                    used = total - left
+                    percentage = round((used / total) * 100, 2) if total > 0 else 0
+                    
+                    # 添加到用户列表
+                    user_points = {
+                        "user_id": i + 1,  # 用户ID从1开始
+                        "left": left,
+                        "total": total,
+                        "used": used,
+                        "percentage": percentage,
+                        "last_update": datetime.now()
+                    }
+                    users_compute_points.append(user_points)
+                    
+                    # 累计总数
+                    total_left += left
+                    total_points += total
+                    
+                    print(f"用户{i+1}计算点信息更新成功: 剩余 {left}, 总计 {total}")
+                    
+                    # 对于第一个用户，获取计算点使用日志
+                    if i == 0:
+                        get_compute_points_log(session, cookies, session_token)
+                else:
+                    print(f"获取用户{i+1}计算点信息失败: {result.get('error', '未知错误')}")
             else:
-                print(f"获取计算点信息失败: {result.get('error', '未知错误')}")
-        else:
-            print(f"获取计算点信息失败，状态码: {response.status_code}")
-    except Exception as e:
-        print(f"获取计算点信息异常: {e}")
+                print(f"获取用户{i+1}计算点信息失败，状态码: {response.status_code}")
+        except Exception as e:
+            print(f"获取用户{i+1}计算点信息异常: {e}")
+    
+    # 更新全局计算点信息（所有用户总和）
+    if users_compute_points:
+        compute_points["left"] = total_left
+        compute_points["total"] = total_points
+        compute_points["used"] = total_points - total_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"所有用户计算点总计: 剩余 {total_left}, 总计 {total_points}")
 
 # 获取计算点使用日志
 def get_compute_points_log(session, cookies, session_token):
@@ -975,7 +1010,8 @@ def dashboard():
         total_tokens=total_tokens,
         compute_points=compute_points,
         compute_points_log=compute_points_log,
-        space_url=SPACE_URL  # 传递空间URL
+        space_url=SPACE_URL,  # 传递空间URL
+        users_compute_points=users_compute_points  # 传递用户计算点信息
     )
 
 

templates/dashboard.html

@@ -554,7 +554,7 @@
                 <div class="card-header">
                     <h2 class="card-title">
                         <span class="card-icon">💰</span>
-                        计算点信息
+                        计算点总计
                     </h2>
                 </div>
                 <div class="status-item">
@@ -635,6 +635,50 @@
             </div>
         </div>
         
+        {% if users_compute_points|length > 0 %}
+        <div class="card">
+            <div class="card-header">
+                <h2 class="card-title">
+                    <span class="card-icon">👥</span>
+                    用户计算点详情
+                </h2>
+            </div>
+            <div class="table-container">
+                <table class="data-table">
+                    <thead>
+                        <tr>
+                            <th>用户</th>
+                            <th>总计算点</th>
+                            <th>已使用</th>
+                            <th>剩余</th>
+                            <th>使用比例</th>
+                        </tr>
+                    </thead>
+                    <tbody>
+                        {% for user in users_compute_points %}
+                        <tr>
+                            <td>用户 {{ user.user_id }}</td>
+                            <td class="compute-points">{{ user.total|int }}</td>
+                            <td class="compute-points">{{ user.used|int }}</td>
+                            <td class="compute-points">{{ user.left|int }}</td>
+                            <td>
+                                <div style="width: 100%; position: relative;">
+                                    <span class="status-value compute-points {% if user.percentage > 80 %}danger{% elif user.percentage > 50 %}warning{% endif %}">
+                                        {{ user.percentage }}%
+                                    </span>
+                                    <div class="progress-container">
+                                        <div class="progress-bar {% if user.percentage > 80 %}danger{% elif user.percentage > 50 %}warning{% endif %}" style="width: {{ user.percentage }}%"></div>
+                                    </div>
+                                </div>
+                            </td>
+                        </tr>
+                        {% endfor %}
+                    </tbody>
+                </table>
+            </div>
+        </div>
+        {% endif %}
+        
         <div class="card">
             <div class="card-header">
                 <h2 class="card-title">