Flask使用Session和上下文處理器保持登錄狀態

(1)Flask如何實現用戶登錄後註銷、用戶登錄後權限不同,頁面不一樣的功能呢?

           我們可以使用session來保存用戶登錄的這個狀態,當登錄成功的時候,將當前登錄的username存入session,並設置sessionpermanent屬性爲True(這樣下次打開瀏覽器仍然是登錄的狀態),這一步很簡單,login函數修改如下:

#登錄
@app.route('/login/',methods=['POST','GET'])
def login():
    if request.method=='GET':
        return render_template('login.html')
    else:
        telephone=request.form.get('telephone')
        print(telephone)
        password=request.form.get('password')
        print(password)
        user=User.query.filter(User.telephone==telephone,User.password==password).first()
        print(user)
        if user:
            session['username']=user.username
            session.permanent=True
            return render_template('index.html')
        else:
            return u'用戶名或者密碼錯誤'

          這個時候只是將當前的用戶名存入到session中了,並不能讓前端導航條上的註冊改變爲用戶名,這個時候需要使用上下文處理器來傳遞參數,使用@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('logup') }}">註冊</a></li>
     <li><a href="{{ url_for('login') }}">登錄</a></li>
 {% endif %}

此時登錄成功後就變成下面這樣了:

       上面的代碼,我們把註銷關聯到了logout視圖函數,在這個函數裏使用session.clear()清除session中的內容。

#登出
@app.route('/logout/',methods=['POST','GET'])
def logout():
    session.clear()
    return render_template('index.html')

       整體的前端:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章