shudao-main/shudao-chat-py/create_admin_user.py

#!/usr/bin/env python3
"""创建Admin管理员用户"""
from database import SessionLocal
from models.total import User
import bcrypt

def create_admin_user():
    db = SessionLocal()
    try:
        # 检查Admin用户是否已存在
        existing = db.query(User).filter(User.username == 'Admin').first()
        if existing:
            print("✅ Admin用户已存在")
            print(f"   用户名: {existing.username}")
            print(f"   角色: {existing.role}")
            print(f"   ID: {existing.id}")
            return
        
        # 创建密码哈希
        password = "123456"
        hashed_password = bcrypt.hashpw(password.encode("utf-8"), bcrypt.gensalt()).decode("utf-8")
        
        # 创建管理员用户
        admin_user = User(
            username="Admin",
            password=hashed_password,
            nickname="管理员",
            role="admin",
            status=1,
            points=1000,
            is_deleted=0
        )
        
        db.add(admin_user)
        db.commit()
        db.refresh(admin_user)
        
        print(f"✅ Admin管理员创建成功!")
        print(f"   用户名: {admin_user.username}")
        print(f"   密码: 123456")
        print(f"   角色: {admin_user.role}")
        print(f"   初始积分: {admin_user.points}")
        print(f"   ID: {admin_user.id}")
        
    except Exception as e:
        db.rollback()
        print(f"❌ 创建失败: {e}")
        import traceback
        traceback.print_exc()
    finally:
        db.close()

if __name__ == "__main__":
    create_admin_user()