def register_user():
if current_user.is_authenticated():
flash(
'You are already logged in as %s' % (escape(current_user.username))
)
return redirect(url_for('ui.index'))
form = RegistrationForm()
if form.validate_on_submit():
try:
username = form.username.data
password = form.password.data
email = form.email.data.strip()
if len(email) == 0:
email = None
create_user(username, password, email)
login(username, password)
flash('Registration successful, welcome %s!' % (username),
category='info')
return redirect(url_for('ui.index'))
except ValidationError, ve:
invalids = ','.join([f.title() for f in ve.errors.keys()])
msg = 'Invalid: %s' % (invalids)
flash(escape(msg), category='error')
except ValueError, ve:
flash(escape(ve.message), category='error')
def login_user():
def redirect_url():
forward = safe_redirect_url()
if not forward:
flash("Logged in successfully.", category='info')
return redirect(forward or url_for('ui.index'))
# If you are already logged in, go away!
if not current_user.is_anonymous():
return redirect_url()
if request.method == 'POST':
username = request.form.get('username', '')
if login(username, request.form.get('password', '')):
return redirect_url()
flash("Invalid username/password", category='error')
return render_template("login.html")
