init_user.py

# -*- coding: utf-8 -*-
# init_user.py
"""
初始化用户脚本
用于创建默认管理员账户
"""
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from server import User, Base

# 【请修改】你的数据库配置（与 server.py 保持一致）
SQLALCHEMY_DATABASE_URL = "mysql+pymysql://root:123456@localhost:3306/rag_db"

engine = create_engine(SQLALCHEMY_DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)


def init_db():
    """初始化数据库和默认用户"""
    print("=" * 50)
    print("📊 正在初始化数据库...")
    print("=" * 50)
    
    try:
        # 1. 创建所有表
        Base.metadata.create_all(bind=engine)
        print("✅ 数据库表创建成功")
        
        db = SessionLocal()
        
        # 2. 检查是否存在 admin 用户
        user = db.query(User).filter(User.username == "admin").first()
        if not user:
            # 创建默认管理员（明文密码）
            new_user = User(username="admin", password="123456")
            db.add(new_user)
            db.commit()
            print("✅ 已创建默认管理员账户:")
            print("   用户名: admin")
            print("   密码: 123456")
        else:
            print("⚠️  用户 'admin' 已存在，跳过创建")
        
        # 3. 显示所有用户
        all_users = db.query(User).all()
        if all_users:
            print(f"\n📋 当前数据库中的用户 ({len(all_users)} 个):")
            for u in all_users:
                print(f"   - {u.username} (ID: {u.id})")
        else:
            print("\n⚠️  数据库中没有用户")
        
        db.close()
        print("\n" + "=" * 50)
        print("✅ 初始化完成！")
        print("=" * 50)
        
    except Exception as e:
        print(f"\n❌ 初始化失败: {e}")
        import traceback
        traceback.print_exc()
        return False
    
    return True


if __name__ == "__main__":
    init_db()