34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
"""
|
||
确保默认管理员账号存在：username=admin, password=123456，角色 admin。
|
||
"""
|
||
from app import db
|
||
from app.models.user import User, Role
|
||
from app.utils.auth import hash_password
|
||
|
||
|
||
def ensure_admin_user(admin_username="admin", admin_password="123456", admin_email=None):
|
||
"""若不存在则创建管理员角色及管理员用户。"""
|
||
if not admin_email:
|
||
admin_email = f"{admin_username}@localhost"
|
||
role = Role.query.filter_by(name="admin").first()
|
||
if not role:
|
||
role = Role(name="admin", description="Administrator")
|
||
db.session.add(role)
|
||
db.session.flush()
|
||
user = User.query.filter_by(username=admin_username).first()
|
||
if not user:
|
||
user = User(
|
||
username=admin_username,
|
||
email=admin_email,
|
||
password_hash=hash_password(admin_password),
|
||
role_id=role.id,
|
||
is_active=True,
|
||
)
|
||
db.session.add(user)
|
||
else:
|
||
if user.role_id != role.id:
|
||
user.role_id = role.id
|
||
if not user.is_active:
|
||
user.is_active = True
|
||
db.session.commit()
|