def create_superuser_role():
try:
superuser = User.objects.get(email='*****@*****.**')
role = Role(name='superuser', description='admin of all admins')
role.save()
superuser.update(push__roles=role)
except DoesNotExist:
print(
"superuser account must exist in order to create a super admin role"
)
except NotUniqueError:
print("a superuser role already exist")
def role_add():
form = RoleForm()
if form.validate_on_submit():
data = form.data
role = Role(name=data["name"],
auths=",".join(map(lambda v: str(v), data["auths"])))
db.session.add(role)
db.session.commit()
flash("添加角色成功！", "ok")
oplog = Oplog(admin_id=session["admin_id"],
ip=request.remote_addr,
reason="添加新角色：%s" % data["name"])
db.session.add(oplog)
db.session.commit()
return redirect(url_for("admin.role_add"))
return render_template("role_add.html", form=form)
