Upload 70 files

config.py

@@ -28,7 +28,11 @@ PLATFORMS = [
     {"id": "anthropic", "name": "Anthropic"},
     {"id": "openai", "name": "OpenAI"},
     {"id": "google", "name": "Google"},
-    {"id": "deepseek", "name": "Deepseek"}
+    {"id": "deepseek", "name": "Deepseek"},
+    {"id": "siliconflow", "name": "SiliconFlow"},
+    {"id": "xai", "name": "xAI"},
+    {"id": "groq", "name": "Groq"},
+    {"id": "openrouter", "name": "OpenRouter"}
 ]
 
 # 平台标签样式
@@ -52,16 +56,40 @@ PLATFORM_STYLES = {
         "background-color": "rgba(77, 107, 254, 0.1)",
         "border-color": "rgba(77, 107, 254, 0.3)",
         "color": "#3246d3"
+    },
+    "siliconflow": {
+        "background-color": "rgba(124, 58, 237, 0.1)",
+        "border-color": "rgba(124, 58, 237, 0.3)",
+        "color": "#6429c8"
+    },
+    "xai": {
+        "background-color": "rgba(90, 90, 90, 0.1)",
+        "border-color": "rgba(90, 90, 90, 0.3)",
+        "color": "#5A5A5A"
+    },
+    "groq": {
+        "background-color": "rgba(255, 95, 31, 0.1)",
+        "border-color": "rgba(255, 95, 31, 0.3)",
+        "color": "#FF5F1F"
+    },
+    "openrouter": {
+        "background-color": "rgba(61, 88, 171, 0.1)",
+        "border-color": "rgba(61, 88, 171, 0.3)",
+        "color": "#3D58AB"
     }
 }
 
 # API调用节流控制配置
 PLATFORM_LIMITS = {
-    'openai': 50,
-    'anthropic': 50,  
-    'google': 50,    
-    'deepseek': 50,   
-    'default': 10   
+    'openai': 100,
+    'anthropic': 100,  
+    'google': 500,    
+    'deepseek': 50,
+    'siliconflow': 50,
+    'xai': 50,
+    'groq': 50,
+    'openrouter': 50,
+    'default': 50
 }
 
 # 请求节流时间窗口（秒）

core/api/xai.py

@@ -0,0 +1,69 @@
+"""
+xAI API相关功能模块 - 提供xAI API的调用和验证功能
+"""
+import requests
+
+def validate_api_key(api_key):
+    """
+    验证xAI API密钥是否有效
+    
+    Args:
+        api_key (str): 需要验证的xAI API密钥
+    
+    Returns:
+        dict: 包含验证结果的字典，包括:
+            - success (bool): API密钥验证成功为True，失败为False
+            - return_message (str): 验证响应的具体消息，成功时为"200: Success!"，失败时为错误详情
+            - balance (float): 账户余额，固定为0
+            - states (str): 账户状态，xAI没有Tier等级，默认为空字符串
+    """
+    url = "https://api.x.ai/v1/chat/completions"
+    
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    
+    payload = {
+        "model": "grok-2-1212",
+        "max_tokens": 1,
+        "messages": [
+            {"role": "user", "content": "Hello"}
+        ]
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        
+        if response.status_code == 200:
+            return {
+                "success": True,
+                "return_message": "200: Success!",
+                "balance": 0,
+                "states": ""
+            }
+        else:
+            return_message = ""
+            try:
+                error_data = response.json()
+                if "error" in error_data:
+                    return_message = error_data["error"].get("message", "")
+                    return_message = f"{response.status_code}: {return_message}"
+            except:
+                return_message = f"{response.status_code}: {response.text or f'HTTP错误'}"
+                
+            return {
+                "success": False,
+                "return_message": return_message,
+                "balance": 0,
+                "states": ""
+            }
+    except Exception as e:
+        error_str = str(e)
+        print(f"验证API密钥时出错: {error_str}")
+        return {
+            "success": False,
+            "return_message": f"500: 请求异常: {error_str}",
+            "balance": 0,
+            "states": ""
+        }

core/update.py

@@ -4,6 +4,7 @@ API密钥更新模块 - 提供API密钥的验证和更新功能
 import json
 import os
 import sqlite3
+import pytz
 from datetime import datetime
 from core.api_manager import get_api_manager
 from utils.db import get_db_connection
@@ -60,8 +61,8 @@ def update(key_id):
         except Exception as e:
             return {"success": False, "message": f"验证API密钥时出错: {str(e)}"}
         
-        # 当前时间
-        current_time = datetime.now().isoformat()
+        # 当前时间 - 使用亚洲/上海时区 (UTC+8)
+        current_time = datetime.now(pytz.timezone('Asia/Shanghai')).isoformat()
         
         # 将布尔值转换为整数
         success_int = 1 if result.get("success", False) else 0

routes/web.py

@@ -1,6 +1,7 @@
 """
 Web路由模块 - 处理所有页面请求和身份验证
 """
+import re # 导入正则表达式模块
 from flask import Blueprint, render_template, request, redirect, url_for, make_response, session
 from datetime import datetime
 from config import ADMIN_PASSWORD, PLATFORMS, PLATFORM_STYLES, TOKEN_EXPIRY_DAYS
@@ -10,10 +11,50 @@ from models.api_key import ApiKeyManager
 # 创建Web蓝图
 web_bp = Blueprint('web', __name__)
 
+# 辅助函数：为API密钥生成排序键
+def get_sort_key(key):
+    """
+    生成用于排序的元组 (balance, platform_priority)。
+    - balance: 主要排序键，降序。
+    - platform_priority: 次要排序键，仅在 balance 相同时生效，降序。
+        - Google: paid=1, free=0
+        - OpenAI/Claude: t5=5, t4=4, ..., t1=1, 其他=0
+    """
+    balance = key.get('balance', 0)
+    platform_priority = 0
+    platform = key.get('platform', '').lower()
+    name = key.get('name', '').lower() # 用于 OpenAI/Claude tier 判断
+    states = key.get('states', '').lower() # 获取 states 字段并转小写，用于 Google 判断
+
+    if platform == 'google':
+        if 'paid' in states: # 检查 states 字段是否包含 'paid' (已转小写)
+            platform_priority = 1 # paid 优先
+    elif platform in ['openai', 'anthropic']: # 假设 Claude 的 platform id 是 'anthropic'
+        match = re.search(r't(\d)', name)
+        if match:
+            try:
+                tier = int(match.group(1))
+                if 1 <= tier <= 5: # 假设 tier 在 1-5 之间
+                    platform_priority = tier
+            except ValueError:
+                pass # 如果转换失败，保持 priority 为 0
+
+    # 返回元组，Python 的 sorted 会按元组顺序比较
+    # 因为 sorted 默认升序，我们需要对 balance 和 priority 都取反来实现降序效果
+    # 或者直接在 sorted 中用 reverse=True，然后只对 platform_priority 取反（或用正数表示优先级）
+    # 这里我们用 reverse=True，所以 balance 自然降序，platform_priority 越大越优先
+    return (balance, platform_priority)
+
 @web_bp.route('/')
 def index():
     """首页 - 显示API密钥管理界面"""
     api_keys_data = ApiKeyManager.get_all_keys()
+    all_keys_list = api_keys_data.get("api_keys", [])
+
+    # 按 balance 降序排序密钥
+    # 注意：确保 balance 存在且为数字类型，否则排序可能出错。使用 .get('balance', 0) 提供默认值。
+    # 使用新的排序逻辑
+    sorted_keys = sorted(all_keys_list, key=get_sort_key, reverse=True)
 
     # 按平台分组密钥
     grouped_keys = {}
@@ -24,7 +65,7 @@ def index():
         }
     
     # 将密钥加入对应的平台组
-    for key in api_keys_data.get("api_keys", []):
+    for key in sorted_keys: # 使用排序后的列表进行分组
         platform_id = key.get("platform", "other").lower()
         if platform_id not in grouped_keys:
             platform_id = "other"

static/js/api-key-manager/core.js

@@ -10,19 +10,21 @@ function initApiKeyManager() {
     // 初始化平台ID列表
     this.platformIds = platforms.map(platform => platform.id);
     
-    // 从localStorage读取平台展开状态，如果有的话
-    const savedPlatformStates = localStorage.getItem('platformStates');
+    // 从localStorage读取当前视图状态，如果有的话
+    const savedView = localStorage.getItem('keyView');
+    if (savedView === 'valid' || savedView === 'invalid') {
+        this.currentView = savedView;
+    }
+    
+    // 从localStorage读取对应视图的平台展开状态，如果有的话
+    const stateKey = `platformStates_${this.currentView}`;
+    const savedPlatformStates = localStorage.getItem(stateKey);
     const parsedPlatformStates = savedPlatformStates ? JSON.parse(savedPlatformStates) : {};
     
     // 从localStorage读取平台筛选状态，如果有的话
     const savedPlatformFilters = localStorage.getItem('platformFilters');
     const parsedPlatformFilters = savedPlatformFilters ? JSON.parse(savedPlatformFilters) : {};
     
-    // 从localStorage读取当前视图状态，如果有的话
-    const savedView = localStorage.getItem('keyView');
-    if (savedView === 'valid' || savedView === 'invalid') {
-        this.currentView = savedView;
-    }
     
     // 初始化平台状态，优先使用保存的状态
     platforms.forEach(platform => {

static/js/api-key-manager/main-manager.js

@@ -140,10 +140,29 @@ function apiKeyManager() {
         // 切换当前视图（有效/无效密钥）
         switchView(view) {
             if (this.currentView !== view) {
+                // 保存当前视图的平台展开状态
+                const currentStateKey = `platformStates_${this.currentView}`;
+                localStorage.setItem(currentStateKey, JSON.stringify(this.platformStates));
+                
+                // 切换视图
                 this.currentView = view;
-                // 保存用户的选择到localStorage
+                
+                // 保存用户的视图选择到localStorage
                 localStorage.setItem('keyView', view);
                 
+                // 加载新视图的平台展开状态
+                const newStateKey = `platformStates_${view}`;
+                const savedPlatformStates = localStorage.getItem(newStateKey);
+                if (savedPlatformStates) {
+                    this.platformStates = JSON.parse(savedPlatformStates);
+                } else {
+                    // 如果没有保存过这个视图的状态，使用默认值（全部展开）
+                    const platforms = this.getPlatforms();
+                    platforms.forEach(platform => {
+                        this.platformStates[platform.id] = true;
+                    });
+                }
+                
                 // 重置选择状态
                 this.selectedKeys = [];
                 this.selectedPlatforms = [];

static/js/api-key-manager/platform-utils.js

@@ -7,8 +7,9 @@
 function togglePlatform(platformId) {
     this.platformStates[platformId] = !this.platformStates[platformId];
     
-    // 保存展开状态到localStorage
-    localStorage.setItem('platformStates', JSON.stringify(this.platformStates));
+    // 保存展开状态到localStorage - 分开存储不同视图的状态
+    const stateKey = `platformStates_${this.currentView}`;
+    localStorage.setItem(stateKey, JSON.stringify(this.platformStates));
 }
 
 // 获取平台API密钥数量