diff --git a/hello.py b/hello.py index 213ea3b..d54828d 100644 --- a/hello.py +++ b/hello.py @@ -1,4 +1,4 @@ -from flask import Flask, render_template +from flask import Flask, render_template, session, redirect, url_for from flask_bootstrap import Bootstrap from flask_moment import Moment from flask_wtf import FlaskForm @@ -30,9 +30,8 @@ def internal_server_error(e): @app.route('/', methods=['GET', 'POST']) def index(): - name = None form = NameForm() if form.validate_on_submit(): - name = form.name.data - form.name.data = '' - return render_template('index.html', form=form, name=name) + session['name'] = form.name.data + return redirect(url_for('index')) + return render_template('index.html', form=form, name=session.get('name'))