def v_user_create():
form = UserCreateForm()
form.roles.choices = possible_roles
if request.method == 'POST' and form.validate_on_submit():
a_user = UserApi()
input_data = {
'email': form.email.data,
'password': form.password.data,
'roles': form.roles.data,
'username': form.email.data
}
try:
new_user = a_user.create(input_data)
except DatabaseItemAlreadyExists as e:
flash('A user called {0} already exists.'.format(input_data['email']))
return render_template('admin/user/create.html', form=form)
except RequiredAttributeMissing as e:
flash('A required form element was not submitted: {0}'.format(e))
return render_template('admin/user/create.html', form=form)
except Exception as e: # Remove this after debugging
#flash('An unexpected error occurred: {0}'.format(e))
flash('An unexpected error occurred.')
return render_template('admin/user/create.html', form=form)
else:
return redirect(url_for('.v_user_list'))
return render_template('admin/user/create.html', form=form)
def v_user_edit(user_id):
form = UserModifyForm()
a_user = UserApi()
form.roles.choices = possible_roles
try:
existing_user = a_user.read(user_id)
except DatabaseItemDoesNotExist:
flash('A user with id {0} does not exist.'.format(user_id))
return redirect(url_for('.v_user_list'))
if request.method == 'POST' and form.validate_on_submit():
input_data = {
'email': form.email.data,
'password': form.password.data,
'roles': form.roles.data
}
##
# We very much dislike empty passwords, so we assume that, if the password field
# is empty, the user didn't want to change it.
if input_data['password'] == '' or input_data['password'] is None:
update_password = False
else:
update_password = True
try:
edited_user = a_user.update(user_id, input_data, update_password)
except DatabaseItemDoesNotExist as e:
flash('No user with id {0}'.format(user_id))
return redirect(url_for('.v_user_list'))
except RequiredAttributeMissing as e:
flash('A required form element was not submitted: {0}'.format(e))
return render_template('admin/user/edit.html', form=form, user_id=user_id)
except Exception as e:
flash('An unexpected error occurred: {0}'.format(e))
#flash('An unexpected error occurred.')
return render_template('admin/user/edit.html', form=form, user_id=user_id)
else:
return redirect(url_for('.v_user_list'))
else:
##
# Add the data from the existing user for the edit form. This must be done after validate_on_submit()
# or it will overwrite the data from the submitted form.
form.email.default = existing_user.email
roles_default = []
for role in existing_user.roles:
roles_default.append(str(role.id))
form.roles.default = roles_default
# http://stackoverflow.com/questions/5519729/wtforms-how-to-select-options-in-selectmultiplefield
form.process()
return render_template('admin/user/edit.html', form=form, user_id=user_id)
def v_login():
form = LoginForm()
a_user = UserApi()
if request.method == 'POST' and form.validate_on_submit():
try:
user = a_user.get_by_user(form.email.data)
except DatabaseItemDoesNotExist:
# User does not exist
flash('Invalid username or password.')
else:
if user.verify_password(form.password.data):
login_user(user, form.remember_me.data)
return redirect(request.args.get('next') or url_for('.v_index'))
else:
flash('Invalid username or password.')
##
# next=request.args.get('next') must be embedded in the <form action='admin/login?next=next'>, or
# otherwise the POST request (when you submit the form) will not include the "next" bit.
return render_template('admin/login.html', form=form, next=request.args.get('next'))
def v_login():
form = LoginForm()
a_user = UserApi()
if request.method == 'POST' and form.validate_on_submit():
try:
user = a_user.get_by_user(form.email.data)
except DatabaseItemDoesNotExist:
# User does not exist
flash('Invalid username or password.')
else:
if user.verify_password(form.password.data):
login_user(user, form.remember_me.data)
return redirect(
request.args.get('next') or url_for('.v_index'))
else:
flash('Invalid username or password.')
##
# next=request.args.get('next') must be embedded in the <form action='admin/login?next=next'>, or
# otherwise the POST request (when you submit the form) will not include the "next" bit.
return render_template('admin/login.html',
form=form,
next=request.args.get('next'))
def v_user_delete(user_id):
form = UserDeleteForm()
a_user = UserApi()
try:
existing_user = a_user.read(user_id)
except DatabaseItemDoesNotExist as e:
flash('No user with id {0}'.format(user_id))
return redirect(url_for('.v_user_list'))
except Exception as e:
flash('An unexpected error occurred: {0}'.format(e))
# flash('An unexpected error occurred.')
return redirect(url_for('.v_user_list'))
if request.method == 'POST' and form.validate_on_submit():
if a_user.delete(user_id) is True:
flash('User {0} deleted'.format(existing_user.email))
return redirect(url_for('.v_user_list'))
else:
flash('Unable to delete user {0}'.format(existing_user.email))
return render_template('admin/user/delete.html', form=form, user_id=user_id, user_name=existing_user.email)
return render_template('admin/user/delete.html', form=form, user_id=user_id, user_name=existing_user.email)
def v_user_list():
a_user = UserApi()
l_users = a_user.list()
return render_template('admin/user/list.html', users=l_users)
