【Flask项目】首页用户显示与退出 # 10

如图所见，首页用户的展示应当是 "登录 / 注册" 与 "用户信息  退出" 二选其一

对此，做出一些操作，使存在状态保持时，显示用户信息。

不存在session时，显示  "登录 / 注册"

将首页相关的视图函数都写在index对应的views里

from . import blue_index
from flask import render_template, current_app, session, abort
from info.models import User


@blue_index.route("/")
def index():
    """
        首页相关
            右上角个人信息
            1.获取session信息
            2.利用id查询数据库中信息
            3.返回信息到前端页面
    :return:
    """
    # 1.获取session信息(id)
    id = session.get("id")
    user = None
    
    # 2.利用id查询数据库中信息
    try:
        user = User.query.filter_by(id=id).first()
    except Exception as e:
        current_app.logger.error(e)
        abort(400)
    content = {
        "user": user
    }
    
    # 3.返回信息到前端页面
    return render_template("index.html", content=content)

这里使用的是前后端不分离的写法：

            {% if content.user %}
                
                    
                    {{ content.user.nick_name }}
                    退出
                
            {% else %}
                
                    登录 / 注册
                
            {% endif %}

将登出相关的视图函数都写在passport对应的views里

from flask import jsonify, session
from info.response_code import RET
from . import blue_passport

@blue_passport.route("/logout", methods=["GET"])
def logout():
    """
        登出相关
        1.删除session信息
        2.返回结果
    :return:
    """
    session.pop("id", None)
    session.pop("mobile", None)
    session.pop("nickName", None)
    return jsonify(errno=RET.OK, errmsg="退出成功")

// 退出登录
function logout() {
    // $.ajax({
    //     url:'/passport/logout',
    //     type:'get'
    // });

    $.get('/passport/logout', function (response) {
        if (response.errno == '0') {
            // 退出登录成功
            location.reload();
        } else {
            alert(response.errmsg);
        }
    })
}