shudao-main/shudao-chat-py/reset_admin_password.py

#!/usr/bin/env python3
"""重置Admin用户密码"""
from database import SessionLocal
from models.total import User
import bcrypt

def reset_admin_password():
    db = SessionLocal()
    try:
        # 查找Admin用户
        admin = db.query(User).filter(User.username == 'Admin').first()
        if not admin:
            print("❌ Admin用户不存在")
            return
        
        # 重置密码为123456
        password = "123456"
        hashed_password = bcrypt.hashpw(password.encode("utf-8"), bcrypt.gensalt()).decode("utf-8")
        
        admin.password = hashed_password
        admin.role = "admin"  # 确保是admin角色
        
        db.commit()
        
        print(f"✅ Admin密码重置成功!")
        print(f"   用户名: {admin.username}")
        print(f"   新密码: 123456")
        print(f"   角色: {admin.role}")
        print(f"   ID: {admin.id}")
        
    except Exception as e:
        db.rollback()
        print(f"❌ 重置失败: {e}")
        import traceback
        traceback.print_exc()
    finally:
        db.close()

if __name__ == "__main__":
    reset_admin_password()