def admin_user_page(uid):
"""
Renders the user admin page
"""
user_db = UserDB()
role_db = RoleDB()
if current_user.is_authenticated() and current_user.is_active():
# Gather permissions for currently logged in user
# cur_all_groups = set(user_db.get_users_groups(current_user))
cur_rolemap = role_db.get_roles(current_user)
admin_group = user_db.get_group(mainApp.config["ADMIN_GROUP"])
is_admin = user_db.in_group(current_user, admin_group)
gs = set(user_db.get_all_groups())
avail_roles = role_db.get_all_roles()
avail_roles_by_group = dict()
for g in gs:
avail_roles_by_group[g.id] = list(avail_roles)
cur_roles_by_group = _get_roles_by_group(cur_rolemap, avail_roles_by_group)
user = None
group = []
if uid is None:
uid = current_user.id
# We do this because ptherwise we get a proxy to the object
user = user_db.get_user(current_user.id)
else:
user = user_db.get_user(uid)
if user is None:
return render_template(
"noink_message.html",
state=get_state(),
title=_("User not found!"),
message=_('User with ID "{0}" was not found!'.format(uid)),
)
else:
group = user_db.get_users_groups(user)
all_groups = set(user_db.get_users_groups(user))
rolemap = role_db.get_roles(user)
avail_groups = list(gs - all_groups)
user_roles = set()
for m in rolemap:
user_roles.add(m.role)
roles_by_group = _get_roles_by_group(rolemap, avail_roles_by_group)
#
# DETERMINE PERMISSIONS MORE SPECIFICALLY
#
can_edit_users = False
if admin_group.id in cur_roles_by_group:
for rm in cur_roles_by_group[admin_group.id]:
acts = role_db.get_activities(rm.role)
can_edit_users = acts.get("edit_user")
if is_admin or uid == current_user.id:
if request.method == "POST":
if "form_id" in request.form:
if request.form["form_id"] == "general":
#
# PASSWORD UPDATE
#
if request.form.get("password", "") != "":
password = _check_password(
request.form.get("password", None), request.form.get("pcheck", None)
)
if password != "" and password is not None:
user_db.update_password(user, password)
flash(_("Updated password"))
else:
flash(_("Passwords do not match!"))
#
# NAME UPDATE
#
if request.form.get("name", user.name) != user.name:
new_name = _check_username(request.form.get("name", user.name))
if new_name != user.name and new_name is not None:
user.name = new_name
flash(_("User name updated"))
else:
flash(_('"{0}" user already exists!'.format(new_name)), "error")
#
# FULLNAME & BIO UPDATE
#
user.fullname = request.form.get("fullname", user.fullname)
user.bio = request.form.get("bio", user.bio)
#
# PRIMARY GROUP
#
primary_group = request.form.get("primary_group", None)
if primary_group is not None and primary_group != user.primary_group.name:
if user_db.update_primary(user, primary_group):
flash(_("Updated primary group"))
else:
flash(_("Failed to update primary group"), "error")
#
# ACTIVE
#
active = request.form.get("active", None) == "on"
if isinstance(active, bool):
if active != user.active:
user.active = active
flash(_("User active setting changed"))
#
# UPDATE DATABASE
#
user_db.update_user(user)
elif request.form["form_id"] == "groups":
#
# UPDATE GROUP MEMBERSHIPS
#
if "delete" in request.form:
rm_g = user_db.get_group(int(request.form["delete"]))
if isinstance(rm_g, Group):
if user_db.remove_from_group(user, rm_g):
flash(_('User removed from group "{0}".'.format(rm_g.name)))
group = user_db.get_users_groups(user)
all_groups = set(group)
avail_groups = list(gs - all_groups)
else:
print("HERE")
flash(
_(
'Unable to remove user "{0}" from'
' group "{1}".'.format(user.name, rm_g.name)
)
)
else:
flash(
_(
'Unable to remove user "{0}" from'
'group "{1}".'.format(user.name, request.form["delete"])
)
)
elif "add" in request.form:
add_g = user_db.get_group(request.form.get("add_group", None))
if isinstance(add_g, Group):
user_db.add_to_group(user, add_g)
group = user_db.get_users_groups(user)
all_groups = set(group)
avail_groups = list(gs - all_groups)
if add_g in group:
flash(_('Added user to group "{0}".'.format(add_g.name)))
else:
flash(
_('Problem adding user "{0} to ' 'group "{1}"!'.format(user.name, add_g.name)),
"error",
)
else:
flash(_('Unable to find group "{0}"!'.format(request.form["add"])), "error")
elif request.form["form_id"] == "roles":
#
# UPDATE ROLE MEMBERSHIP
#
if "delete" in request.form:
rm_r = role_db.get_role(int(request.form["delete"]))
g = user_db.get_group(int(request.form.get("group", -1)))
if isinstance(rm_r, Role) and isinstance(g, Group):
rolemap = role_db.get_rolemapping(user, g, rm_r)
if rolemap is not None:
role_db.revoke_role(user, g, rm_r)
roles_by_group[g.id].remove(rolemap)
avail_roles_by_group[g.id].append(rm_r)
flash(_('Role "{0}" revoked.'.format(rm_r.name)))
else:
flash(_("Role not assigned to user."), "error")
else:
flash(_("Problem finding role or group.", "error"))
elif "add" in request.form:
g = user_db.get_group(int(request.form.get("group", -1)))
r = role_db.get_role(request.form.get("add_role", None))
if isinstance(r, Role) and isinstance(g, Group):
role_db.assign_role(user, g, r)
rolemap = role_db.get_rolemapping(user, g, r)
if r in avail_roles_by_group[g.id]:
avail_roles_by_group[g.id].remove(r)
if g.id not in roles_by_group:
roles_by_group[g.id] = []
if rolemap not in roles_by_group[g.id]:
roles_by_group[g.id].append(rolemap)
flash(
_(
'Role "{0}" assigned for user "{1}"'
' with group "{2}".'.format(r.name, user.name, g.name)
)
)
else:
flash(_("Problem finding role or group.", "error"))
else:
return render_template(
"noink_message.html",
state=get_state(),
title=_("Form error"),
message=_(
"There was a problem identifying form "
"elements. If this problem persists, contact"
" your site administrator"
),
)
# render the admin user page for uid user
return render_template(
"admin_user.html",
state=get_state(),
user=user,
groups=group,
avail_groups=avail_groups,
is_admin=is_admin,
role_map=rolemap,
avail_roles=avail_roles,
roles_by_group=roles_by_group,
title=_("User Account"),
avail_roles_by_group=avail_roles_by_group,
can_edit_users=can_edit_users,
submit_button=_("Update"),
)
return _not_auth()
def admin_role_page(rid):
"""
Renders the role admin page
"""
user_db = UserDB()
role_db = RoleDB()
if current_user.is_authenticated() and current_user.is_active():
is_admin = user_db.in_group(current_user, mainApp.config['ADMIN_GROUP'])
all_activities = set()
for m in role_db.get_roles(current_user):
acts = role_db.get_activities(m.role_id)
for act in acts:
if acts[act]:
all_activities.add(act)
can_view_roles = 'view_roles' in all_activities
can_edit_roles = 'edit_roles' in all_activities
if is_admin or can_view_roles:
if rid is None:
if request.method == 'POST':
if 'delete' in request.form:
rids = request.form.getlist('select')
for rid in rids:
try:
role_db.delete_role(int(rid))
flash(_('Role with ID "{0}" deleted'.format(
rid)))
except RoleNotFound:
flash(_('"{0}" role id not found!'.format(
rid)), 'error')
elif 'new' in request.form:
return redirect(url_for('admin_role.admin_new_role'))
roles = role_db.get_all_roles()
return render_template('list_roles.html', roles=roles,
state=get_state(), can_view_roles=can_view_roles,
can_edit_roles=can_edit_roles, title=_('All Roles'),
delete_button=_('Delete'), new_button=_('New'),
cancel_button=_('Cancel'), activities=activities,
del_title=_('Delete Roles(s)'),
del_warn=_('Deleting roles is a permanent action. '\
'Are you sure?'))
else:
if request.method == "POST":
if 'cancel' in request.form:
return redirect(url_for('admin_role.admin_role_page'))
elif 'submit' in request.form:
role = role_db.get_role(rid)
if role is not None:
role.name = request.form.get('role_name',
role.name)
role.description = request.form.get('description',
role.description)
updated_acts = request.form.getlist('activities')
ract = get_activity_dict(False)
for a in updated_acts:
ract[a] = True
role = role_db.update_temp_role_activities(
role, ract)
role_db.update_role(role)
return redirect(url_for(
'admin_role.admin_role_page'))
role = role_db.get_role(rid)
if role is not None:
return render_template('admin_role.html', role=role,
state=get_state(), title=_('Edit Role'),
cancel_button=_('Cancel'), submit_button=_('Submit'),
can_edit_roles=True, activities=activities)
else:
return _not_auth()
else:
return _not_auth()