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")