Flask学习（二）：Flash案例代码修改

案例中的原始代码为：

@app.route('/', methods=['GET', 'POST'])
def index():
    form = NameForm()
    if form.validate_on_submit():
        old_name = session.get('name')
        if old_name is not None and old_name != form.name.data:
            flash('Looks like you have changed your name!')
        session['name'] = form.name.data
        return redirect(url_for('index'))
    return render_template('index.html', form=form, name=session.get('name'))

但是我使用该代码重新打开一个浏览器，保证原先session已经不存在，然后打开主页之后submit之后会直接抛出异常
说name不存在，查看我们代码是取得session[‘name’]，所以我直接输出session这个dic的所有key，

print({x for x in session.keys()})

当第一次输出的时候之后一个{‘csrf_token’}，也就是说这时候session这个dic里面没有name这个key，所以直接去获取值得话就会报错。因此，需要修改代码，进来的时候需要判断session里面是否有name，如果没有的话则不能取old_name，需要先设置session[‘name’]的值，然后old_name再去取值。代码如下：

def index():
    name = None
    form = NameForm()
    if form.validate_on_submit():
        if 'name' in session:
            old_name = session['name']
            if old_name is not None and old_name != form.name.data:
                flash('Look like you have changed your name')
            elif old_name is None:
                session['name'] = form.name.data
        else:
            session['name'] = form.name.data
            old_name = session['name']
        return redirect(url_for('index'))
    return render_template('index.html', form=form, name=session.get('name'))