config.py

"""
配置文件 - 存储应用的所有配置参数
"""
import os
import secrets

# 基本目录配置
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DATA_DIR = os.path.join(BASE_DIR, 'data')
os.makedirs(DATA_DIR, exist_ok=True)

# 数据文件路径
API_KEYS_FILE = os.path.join(DATA_DIR, 'api_keys.json')  # 保留以供兼容性
AUTH_FILE = os.path.join(DATA_DIR, 'auth_tokens.json')

# SQLite数据库配置
DATABASE_PATH = os.path.join(DATA_DIR, 'api_keys.db')

# 应用密钥配置
SECRET_KEY = os.environ.get('SECRET_KEY', secrets.token_hex(16))
ADMIN_PASSWORD = os.environ.get('PASSWORD', '123456')

# 认证配置
TOKEN_EXPIRY_DAYS = 30

# 支持的平台
PLATFORMS = [
    {"id": "anthropic", "name": "Anthropic"},
    {"id": "openai", "name": "OpenAI"},
    {"id": "google", "name": "Google"},
    {"id": "deepseek", "name": "Deepseek"}
]

# 平台标签样式
PLATFORM_STYLES = {
    "anthropic": {
        "background-color": "rgba(217, 119, 87, 0.1)",
        "border-color": "rgba(217, 119, 87, 0.3)",
        "color": "#c25032"
    },
    "openai": {
        "background-color": "rgba(16, 163, 127, 0.1)",
        "border-color": "rgba(16, 163, 127, 0.3)",
        "color": "#0e8c6b"
    },
    "google": {
        "background-color": "rgba(28, 125, 255, 0.1)",
        "border-color": "rgba(28, 125, 255, 0.3)",
        "color": "#0051c3"
    },
    "deepseek": {
        "background-color": "rgba(77, 107, 254, 0.1)",
        "border-color": "rgba(77, 107, 254, 0.3)",
        "color": "#3246d3"
    }
}

# API调用节流控制配置
PLATFORM_LIMITS = {
    'openai': 30,
    'anthropic': 30,  
    'google': 30,    
    'deepseek': 50,   
    'default': 10   
}

# 请求节流时间窗口（秒）
TIME_WINDOW = 10