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)