修改Flask-Login @Login_Required修飾器

Question

我想更改默認的Flask-Login @login_required，以便它將檢查數據庫是否禁止用戶。

我知道我可以創建一個自定義的裝飾器來檢查我的數據庫並將其添加到我的所有視圖中，但是由於我擁有的視圖的數量，這將不太有用。這將不方便。

這可能嗎？謝謝。

Answer 1

您可以使用下一個方法：

from flask.ext.login import LoginManager 

@login_manager.user_loader 
def load_user_if_not_banned(user_id):   
    user = ... # Load user from DB 
    return user if not user.is_banned else None 

Answer 2

您可以實現自己的login_required裝飾這是相當簡單的實際。這篇文章https://medium.com/@nguyenkims/python-decorator-and-flask-3954dd186cda給出了一些關於如何實現這個裝飾器的僞代碼。