import os
from sqlalchemy.orm import Session
from app.database.database import get_db, engine, Base
from app.models.user import User

# 设置测试环境变量
os.environ["TESTING"] = "True"

# 创建所有表
Base.metadata.create_all(bind=engine)

# 获取数据库会话
db = next(get_db())

# 检查是否已经存在 admin 用户
existing_admin = db.query(User).filter(User.username == "admin").first()
if not existing_admin:
    # 创建 admin 用户，使用明文密码
    admin_user = User(
        username="admin",
        password="admin",
        department="IT",
        role="admin"
    )
    db.add(admin_user)
    db.commit()
    print("Admin user created successfully!")
else:
    print("Admin user already exists!")

# 关闭数据库会话
db.close()