def welcome():
if current_user.username:
return redirect(url_for('user.settings'))
form = WelcomeForm()
if form.validate_on_submit():
current_user.username = form.username.data
db.session.commit()
flash(
"Sign up is complete, you can now enjoy our serveices. Thank you.")
return redirect(url_for('user.settings'))
return render_template('user/welcome.html', form=form)
def welcome():
if current_user.username:
flash('You already picked a username.', 'warning')
return redirect(url_for('user.settings'))
form = WelcomeForm()
if form.validate_on_submit():
current_user.username = request.form.get('username')
current_user.save()
flash('Sign up is complete, enjoy our services.', 'success')
return redirect(url_for('user.settings'))
return render_template('user/welcome.html', form=form)