目前我们的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 %}
<li><a href="#">用户:{{ login_user }}</a></li>
<li><a href="{{ url_for('logout') }}">注销</a></li>
{% else %}
<li><a href="{{ url_for('login') }}">登录</a></li>
<li><a href="{{ url_for('register') }}">注册</a></li>
{% endif %}
此时,登录成功后,导航条就变成如下样式了:
上面的代码,我们把注销关联到了一个logout
视图函数,这个也很好写,用session
的clear
方法清除session
中的内容,然后跳转回首页:
@app.route('/logout/')
def logout():
session.clear()
return redirect(url_for('home'))
用户名的那部分,我们后续设计一个可以更改头像和修改密码的页面,点击用户名跳转过去,这里就先写一个#
。