目前我们的demo已经完成了注册和登录功能,当登录成功之后,页面跳转到首页,此时我们需要保存用户的登录状态,不管页面如何跳转,除非用户自己注销登录。
如之前所说,我们可以用session来保存用户登录的这个状态,当登录验证成功的时候,将当前登录的username存入session,并设置session的permanent属性为True(这样下次打开浏览器仍然是登录的状态),这一步很简单,login函数修改如下:
@app.route('/login/', methods=['GET', 'POST'])
def login():
#...省略...
if '成功' in message:
session['username'] = username
session.permanent = True
return redirect(url_for('home'))
#...省略...
登录成功之后,发现导航条右边的文字还是这个鬼样子:
我们需要修改它,设计成登录成功之后,这一部分显示用户名和注销按钮。用 9.注册和登录功能实现(2)—— 注册与登录的错误提示 提到的@app.context_processor上下文管理器,把当期登录的用户名传递给html:
@app.context_processor
def my_context_processor():
user = session.get('username')
if user:
return {'login_user': user}
return {}
然后在base.html中修改导航条原先登录和注册的部分如下:
{% if login_user %}
用户:{{ login_user }} 注销{% else %}
登录 注册{% endif %}
此时,登录成功后,导航条就变成如下样式了:
上面的代码,我们把注销关联到了一个logout视图函数,这个也很好写,用session的clear方法清除session中的内容,然后跳转回首页:
@app.route('/logout/')
def logout():
session.clear()
return redirect(url_for('home'))
用户名的那部分,我们后续设计一个可以更改头像和修改密码的页面,点击用户名跳转过去,这里就先写一个#。