我們來解決上一文章說到的,為登錄不能點擊發(fā)布問答功能
(一)先寫一個裝飾器
裝飾器教程參考廖雪峰教程 http://t.cn/RK0SaGl

from functools import wraps #登錄限制裝飾器 def login_required(func):@wraps(func)def wrapper(*args,**kwargs):if session.get('user_id'):return func(*args,**kwargs)else:return redirect(url_for('login'))return wrapper

(二).然后在發(fā)布問答函數(shù)修改,仔細看,只是添加了@login_required

@app.route('/question') @login_required def question():if request.method == 'GET':return render_template('question.html')else:pass

現(xiàn)在的函數(shù)相當于

question = login_required(question) = wrapper

(1)如果現(xiàn)在打印print(run.__name__) 會發(fā)現(xiàn)此時結(jié)果變成了wrapper,但這是一種危險的情況,為了解決這種問題,我們導(dǎo)入

from functools import wraps

(2)還有一點def wrapper(*args,**kwargs): 可以傳入的參數(shù)不固定,也就是說當其他函數(shù)使用這個裝飾器的時候,傳入的參數(shù)沒有了限制,你不用擔心因為參數(shù)問題而導(dǎo)致程序出錯.

這樣就實現(xiàn)了限制登錄,當檢測到?jīng)]有user_id時,那么點擊發(fā)布,就不能跳轉(zhuǎn)

總結(jié)

以上是生活随笔為你收集整理的Flask实战2问答平台-登录限制(装饰器)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。