后端开发完成
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user