Delete 密码迁移.py

密码迁移.py

@@ -1,131 +0,0 @@
-# 密码迁移.py
-# ==========================================
-# 🔐 密码迁移脚本 - 一次性执行
-# ==========================================
-# 作用：将现有用户的明文密码批量转换为 SHA256 哈希
-# 执行时机：部署安全更新后，首次运行此脚本
-# 执行方式：python 密码迁移.py
-# ==========================================
-# ⚠️ 注意事项：
-#   1. 此脚本会直接修改 users.json
-#   2. 执行前请备份数据
-#   3. 只需执行一次，执行后可删除此脚本
-#   4. 登录接口已内置自动升级逻辑，此脚本为可选加速方案
-# ==========================================
-
-import os
-import json
-import hashlib
-
-# 密码哈希配置（必须与 安全认证.py 保持一致）
-PASSWORD_SALT = os.environ.get("PASSWORD_SALT", "ComfyUI-Ranking-Salt-v1")
-
-# 数据库路径配置
-if os.environ.get("SPACE_ID"):
-    LOCAL_DB_DIR = "/tmp/local_db_data"
-else:
-    BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-    LOCAL_DB_DIR = os.path.join(BASE_DIR, "cache")
-
-
-def hash_password(password: str) -> str:
-    """将明文密码转换为 SHA256 哈希值"""
-    salted_password = f"{PASSWORD_SALT}{password}"
-    return hashlib.sha256(salted_password.encode("utf-8")).hexdigest()
-
-
-def is_hashed(password: str) -> bool:
-    """判断密码是否已经是哈希格式（64位十六进制）"""
-    if len(password) != 64:
-        return False
-    try:
-        int(password, 16)
-        return True
-    except ValueError:
-        return False
-
-
-def migrate_passwords():
-    """批量迁移所有用户的密码为哈希格式"""
-    users_file = os.path.join(LOCAL_DB_DIR, "users.json")
-    
-    # 检查文件是否存在
-    if not os.path.exists(users_file):
-        print(f"❌ 用户数据文件不存在: {users_file}")
-        print("   请确认数据库路径配置正确")
-        return
-    
-    # 读取用户数据
-    try:
-        with open(users_file, "r", encoding="utf-8") as f:
-            users_db = json.load(f)
-    except Exception as e:
-        print(f"❌ 读取用户数据失败: {e}")
-        return
-    
-    # 统计信息
-    total_users = len(users_db)
-    migrated_count = 0
-    already_hashed_count = 0
-    skipped_count = 0
-    
-    print("=" * 50)
-    print("🔐 开始密码迁移...")
-    print(f"   总用户数: {total_users}")
-    print("=" * 50)
-    
-    # 遍历所有用户
-    for account, user_data in users_db.items():
-        password = user_data.get("password", "")
-        
-        if not password:
-            print(f"⚠️ 跳过: {account} (无密码字段)")
-            skipped_count += 1
-            continue
-        
-        if is_hashed(password):
-            already_hashed_count += 1
-            continue
-        
-        # 执行哈希迁移
-        user_data["password"] = hash_password(password)
-        migrated_count += 1
-        print(f"✅ 已迁移: {account}")
-    
-    # 保存更新后的数据
-    if migrated_count > 0:
-        try:
-            with open(users_file, "w", encoding="utf-8") as f:
-                json.dump(users_db, f, ensure_ascii=False, indent=2)
-            print("\n" + "=" * 50)
-            print("✅ 迁移完成并已保存!")
-        except Exception as e:
-            print(f"\n❌ 保存失败: {e}")
-            return
-    
-    # 输出统计结果
-    print("\n📊 迁移统计:")
-    print(f"   - 总用户数: {total_users}")
-    print(f"   - 本次迁移: {migrated_count}")
-    print(f"   - 已是哈希: {already_hashed_count}")
-    print(f"   - 跳过: {skipped_count}")
-    print("=" * 50)
-    
-    if migrated_count == 0:
-        print("\n✨ 所有密码均已是哈希格式，无需迁移")
-    else:
-        print(f"\n🎉 成功将 {migrated_count} 个用户的密码升级为哈希存储！")
-
-
-if __name__ == "__main__":
-    print("\n" + "=" * 50)
-    print("🔐 ComfyUI-Ranking 密码安全迁移工具")
-    print("=" * 50)
-    print("\n⚠️ 警告: 此操作将修改 users.json 中的密码字段")
-    print("   建议先备份数据文件\n")
-    
-    confirm = "yes"  # 自动确认，无需手动输入
-    if confirm.lower() == "yes":
-        migrate_passwords()
-    else:
-        print("\n❌ 已取消迁移")