52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
初始化数据库，创建默认管理员账号
|
|
"""
|
|
|
|
from sqlalchemy.orm import Session
|
|
from app.models.database import engine, Base, SessionLocal
|
|
from app.models.models import User
|
|
from app.services.user import UserService
|
|
|
|
|
|
def init_db():
|
|
"""初始化数据库"""
|
|
# 创建所有表
|
|
print("创建数据库表...")
|
|
Base.metadata.create_all(bind=engine)
|
|
print("数据库表创建完成")
|
|
|
|
# 创建默认管理员账号
|
|
print("\n创建默认管理员账号...")
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
# 检查是否已存在管理员账号
|
|
admin_user = db.query(User).filter(User.username == "admin").first()
|
|
|
|
if not admin_user:
|
|
# 创建默认管理员账号
|
|
admin_user = User(
|
|
id="user-admin",
|
|
username="admin",
|
|
email="admin@example.com",
|
|
password_hash=UserService.get_password_hash("admin123"),
|
|
role="admin",
|
|
status="active"
|
|
)
|
|
db.add(admin_user)
|
|
db.commit()
|
|
db.refresh(admin_user)
|
|
print("✅ 默认管理员账号创建成功")
|
|
print(f"用户名: admin")
|
|
print(f"密码: admin123")
|
|
else:
|
|
print("⚠️ 管理员账号已存在")
|
|
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
init_db()
|