11.保存登录状态与注销功能

2019年8月15日 120次阅读 来源: Harpsichord1207

目前我们的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'))
    #...省略...

登录成功之后，发现导航条右边的文字还是这个鬼样子：

《11.保存登录状态与注销功能》

我们需要修改它，设计成登录成功之后，这一部分显示用户名和注销按钮。用 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 %}

此时，登录成功后，导航条就变成如下样式了：

《11.保存登录状态与注销功能》

上面的代码，我们把注销关联到了一个logout视图函数，这个也很好写，用session的clear方法清除session中的内容，然后跳转回首页：

@app.route('/logout/')
def logout():
    session.clear()
    return redirect(url_for('home'))

用户名的那部分，我们后续设计一个可以更改头像和修改密码的页面，点击用户名跳转过去，这里就先写一个#。

    原文作者：Harpsichord1207
    原文地址: https://segmentfault.com/a/1190000012894461
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞