You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.3 KiB
38 lines
1.3 KiB
#导入应用创建函数和数据库对象
|
|
from app import create_app, db
|
|
#导入user模型
|
|
from app.models import User
|
|
|
|
def create_admin_user():
|
|
#创建Flask应用对象
|
|
app = create_app()
|
|
# 在应用上下文中执行数据库操作
|
|
with app.app_context():
|
|
# 检查是否已存在用户名为'admin'的管理员用户
|
|
admin = User.query.filter_by(username='admin').first()
|
|
# 如果不存在管理员用户，则创建一个
|
|
if admin is None:
|
|
# 创建管理员用户对象，设置用户名、邮箱和管理员权限
|
|
admin = User(
|
|
username='admin',
|
|
# 管理员用户名
|
|
email='admin@example.com',
|
|
# 管理员邮箱
|
|
is_admin=True
|
|
# 设置管理员权限
|
|
)
|
|
admin.set_password('123456')
|
|
# 为管理员设置密码
|
|
db.session.add(admin)
|
|
# 将管理员用户添加到数据库会话
|
|
db.session.commit()
|
|
# 提交数据库更改
|
|
print('管理员用户创建成功！')
|
|
# 打印创建成功的提示信息
|
|
else:
|
|
print('管理员用户已存在！')
|
|
# 如果管理员已存在，打印提示信息
|
|
|
|
if __name__ == '__main__':
|
|
create_admin_user()
|
|
# 运行脚本 |