mirror of
https://github.com/zhinianboke/xianyu-auto-reply.git
synced 2025-08-02 20:47:35 +08:00
238 lines
8.9 KiB
Python
238 lines
8.9 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
多用户系统迁移脚本
|
||
将历史数据绑定到admin用户，并创建admin用户记录
|
||
"""
|
||
|
||
import sqlite3
|
||
import hashlib
|
||
import time
|
||
from loguru import logger
|
||
|
||
def migrate_to_multiuser():
|
||
"""迁移到多用户系统"""
|
||
print("🔄 开始迁移到多用户系统...")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
# 连接数据库
|
||
conn = sqlite3.connect('xianyu_data.db')
|
||
cursor = conn.cursor()
|
||
|
||
print("1️⃣ 检查数据库结构...")
|
||
|
||
# 检查users表是否存在
|
||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='users'")
|
||
users_table_exists = cursor.fetchone() is not None
|
||
|
||
if not users_table_exists:
|
||
print(" ❌ users表不存在，请先运行主程序初始化数据库")
|
||
return False
|
||
|
||
print(" ✅ users表已存在")
|
||
|
||
# 检查是否已有admin用户
|
||
cursor.execute("SELECT id FROM users WHERE username = 'admin'")
|
||
admin_user = cursor.fetchone()
|
||
|
||
if admin_user:
|
||
admin_user_id = admin_user[0]
|
||
print(f" ✅ admin用户已存在，ID: {admin_user_id}")
|
||
else:
|
||
print("2️⃣ 创建admin用户...")
|
||
|
||
# 获取当前的admin密码哈希
|
||
cursor.execute("SELECT value FROM system_settings WHERE key = 'admin_password_hash'")
|
||
password_hash_row = cursor.fetchone()
|
||
|
||
if password_hash_row:
|
||
admin_password_hash = password_hash_row[0]
|
||
else:
|
||
# 如果没有设置密码，使用默认密码 admin123
|
||
admin_password_hash = hashlib.sha256('admin123'.encode()).hexdigest()
|
||
print(" ⚠️ 未找到admin密码，使用默认密码: admin123")
|
||
|
||
# 创建admin用户
|
||
cursor.execute('''
|
||
INSERT INTO users (username, email, password_hash, is_active, created_at, updated_at)
|
||
VALUES (?, ?, ?, ?, ?, ?)
|
||
''', ('admin', 'admin@localhost', admin_password_hash, True, time.time(), time.time()))
|
||
|
||
admin_user_id = cursor.lastrowid
|
||
print(f" ✅ admin用户创建成功，ID: {admin_user_id}")
|
||
|
||
print("3️⃣ 检查cookies表结构...")
|
||
|
||
# 检查cookies表是否有user_id字段
|
||
cursor.execute("PRAGMA table_info(cookies)")
|
||
columns = [column[1] for column in cursor.fetchall()]
|
||
|
||
if 'user_id' not in columns:
|
||
print(" 🔧 添加user_id字段到cookies表...")
|
||
cursor.execute("ALTER TABLE cookies ADD COLUMN user_id INTEGER")
|
||
print(" ✅ user_id字段添加成功")
|
||
else:
|
||
print(" ✅ user_id字段已存在")
|
||
|
||
print("4️⃣ 迁移历史数据...")
|
||
|
||
# 统计需要迁移的数据
|
||
cursor.execute("SELECT COUNT(*) FROM cookies WHERE user_id IS NULL")
|
||
cookies_to_migrate = cursor.fetchone()[0]
|
||
|
||
if cookies_to_migrate > 0:
|
||
print(f" 📊 发现 {cookies_to_migrate} 个cookies需要绑定到admin用户")
|
||
|
||
# 将所有没有user_id的cookies绑定到admin用户
|
||
cursor.execute("UPDATE cookies SET user_id = ? WHERE user_id IS NULL", (admin_user_id,))
|
||
|
||
print(f" ✅ 已将 {cookies_to_migrate} 个cookies绑定到admin用户")
|
||
else:
|
||
print(" ✅ 所有cookies已正确绑定用户")
|
||
|
||
print("5️⃣ 验证迁移结果...")
|
||
|
||
# 统计各用户的数据
|
||
cursor.execute('''
|
||
SELECT u.username, COUNT(c.id) as cookie_count
|
||
FROM users u
|
||
LEFT JOIN cookies c ON u.id = c.user_id
|
||
GROUP BY u.id, u.username
|
||
''')
|
||
|
||
user_stats = cursor.fetchall()
|
||
print(" 📊 用户数据统计:")
|
||
for username, cookie_count in user_stats:
|
||
print(f" • {username}: {cookie_count} 个cookies")
|
||
|
||
# 检查是否还有未绑定的数据
|
||
cursor.execute("SELECT COUNT(*) FROM cookies WHERE user_id IS NULL")
|
||
unbound_cookies = cursor.fetchone()[0]
|
||
|
||
if unbound_cookies > 0:
|
||
print(f" ⚠️ 仍有 {unbound_cookies} 个cookies未绑定用户")
|
||
else:
|
||
print(" ✅ 所有cookies已正确绑定用户")
|
||
|
||
# 提交事务
|
||
conn.commit()
|
||
print("\n6️⃣ 迁移完成！")
|
||
|
||
print("\n" + "=" * 60)
|
||
print("🎉 多用户系统迁移成功！")
|
||
print("\n📋 迁移总结:")
|
||
print(f" • admin用户ID: {admin_user_id}")
|
||
print(f" • 迁移的cookies数量: {cookies_to_migrate}")
|
||
print(f" • 当前用户数量: {len(user_stats)}")
|
||
|
||
print("\n💡 下一步操作:")
|
||
print(" 1. 重启应用程序")
|
||
print(" 2. 使用admin账号登录管理现有数据")
|
||
print(" 3. 其他用户可以通过注册页面创建新账号")
|
||
print(" 4. 每个用户只能看到自己的数据")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ 迁移失败: {e}")
|
||
if 'conn' in locals():
|
||
conn.rollback()
|
||
return False
|
||
finally:
|
||
if 'conn' in locals():
|
||
conn.close()
|
||
|
||
def check_migration_status():
|
||
"""检查迁移状态"""
|
||
print("\n🔍 检查多用户系统状态...")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
conn = sqlite3.connect('xianyu_data.db')
|
||
cursor = conn.cursor()
|
||
|
||
# 检查users表
|
||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='users'")
|
||
if not cursor.fetchone():
|
||
print("❌ users表不存在，需要先运行主程序初始化数据库")
|
||
return
|
||
|
||
# 检查用户数量
|
||
cursor.execute("SELECT COUNT(*) FROM users")
|
||
user_count = cursor.fetchone()[0]
|
||
print(f"👥 用户数量: {user_count}")
|
||
|
||
# 检查admin用户
|
||
cursor.execute("SELECT id, username, email, is_active FROM users WHERE username = 'admin'")
|
||
admin_user = cursor.fetchone()
|
||
if admin_user:
|
||
print(f"👑 admin用户: ID={admin_user[0]}, 邮箱={admin_user[2]}, 状态={'激活' if admin_user[3] else '禁用'}")
|
||
else:
|
||
print("❌ admin用户不存在")
|
||
|
||
# 检查cookies表结构
|
||
cursor.execute("PRAGMA table_info(cookies)")
|
||
columns = [column[1] for column in cursor.fetchall()]
|
||
if 'user_id' in columns:
|
||
print("✅ cookies表已支持用户隔离")
|
||
|
||
# 统计各用户的cookies
|
||
cursor.execute('''
|
||
SELECT u.username, COUNT(c.id) as cookie_count
|
||
FROM users u
|
||
LEFT JOIN cookies c ON u.id = c.user_id
|
||
GROUP BY u.id, u.username
|
||
ORDER BY cookie_count DESC
|
||
''')
|
||
|
||
user_stats = cursor.fetchall()
|
||
print("\n📊 用户数据分布:")
|
||
for username, cookie_count in user_stats:
|
||
print(f" • {username}: {cookie_count} 个cookies")
|
||
|
||
# 检查未绑定的数据
|
||
cursor.execute("SELECT COUNT(*) FROM cookies WHERE user_id IS NULL")
|
||
unbound_count = cursor.fetchone()[0]
|
||
if unbound_count > 0:
|
||
print(f"\n⚠️ 发现 {unbound_count} 个未绑定用户的cookies，建议运行迁移")
|
||
else:
|
||
print("\n✅ 所有数据已正确绑定用户")
|
||
else:
|
||
print("❌ cookies表不支持用户隔离，需要运行迁移")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 检查失败: {e}")
|
||
finally:
|
||
if 'conn' in locals():
|
||
conn.close()
|
||
|
||
if __name__ == "__main__":
|
||
import sys
|
||
|
||
if len(sys.argv) > 1 and sys.argv[1] == 'check':
|
||
check_migration_status()
|
||
else:
|
||
print("🚀 闲鱼自动回复系统 - 多用户迁移工具")
|
||
print("=" * 60)
|
||
print("此工具将帮助您将单用户系统迁移到多用户系统")
|
||
print("主要功能:")
|
||
print("• 创建admin用户账号")
|
||
print("• 将历史数据绑定到admin用户")
|
||
print("• 支持新用户注册和数据隔离")
|
||
print("\n⚠️ 重要提醒:")
|
||
print("• 迁移前请备份数据库文件")
|
||
print("• 迁移过程中请勿操作系统")
|
||
print("• 迁移完成后需要重启应用")
|
||
|
||
confirm = input("\n是否继续迁移？(y/N): ").strip().lower()
|
||
if confirm in ['y', 'yes']:
|
||
success = migrate_to_multiuser()
|
||
if success:
|
||
print("\n🎊 迁移完成！请重启应用程序。")
|
||
else:
|
||
print("\n💥 迁移失败！请检查错误信息。")
|
||
else:
|
||
print("取消迁移。")
|
||
|
||
print(f"\n💡 提示: 运行 'python {sys.argv[0]} check' 可以检查迁移状态")
|