Upload 安全认证.py

安全认证.py

@@ -25,14 +25,19 @@ import bcrypt
 # ==========================================
 # 🔑 安全密钥配置
 # ==========================================
-# JWT_SECRET: 用于签名 Token，生产环境必须设置环境变量
+# JWT_SECRET: 用于签名 Token，生产环境应设置环境变量
 # PASSWORD_SALT: 密码哈希加盐，增强安全性
-JWT_SECRET = os.environ.get("JWT_SECRET")
-if not JWT_SECRET:
-    raise RuntimeError("必须设置 JWT_SECRET 环境变量")
-PASSWORD_SALT = os.environ.get("PASSWORD_SALT")
-if not PASSWORD_SALT:
-    raise RuntimeError("必须设置 PASSWORD_SALT 环境变量")
+# 降级策略：环境变量未配置时使用内置默认值（适用于单实例部署）
+_DEFAULT_JWT_SECRET = "ComfyUI-Ranking-JWT-Secret-v2-beta-2026"
+_DEFAULT_PASSWORD_SALT = "ComfyUI-Ranking-Salt-v1"
+
+JWT_SECRET = os.environ.get("JWT_SECRET") or _DEFAULT_JWT_SECRET
+PASSWORD_SALT = os.environ.get("PASSWORD_SALT") or _DEFAULT_PASSWORD_SALT
+
+if JWT_SECRET == _DEFAULT_JWT_SECRET:
+    print("⚠️ [安全认证] JWT_SECRET 未设置环境变量，使用内置默认值（建议生产环境配置独立密钥）")
+if PASSWORD_SALT == _DEFAULT_PASSWORD_SALT:
+    print("⚠️ [安全认证] PASSWORD_SALT 未设置环境变量，使用内置默认值")
 
 # Token 有效期配置（单位：秒）
 TOKEN_EXPIRE_SECONDS = 7 * 24 * 60 * 60  # 默认7天（向后兼容）