flask-login不能在Blueprint对象中使用?

use*_*095 8 blueprint flask flask-login

我有关于烧瓶登录和蓝图的问题.

admin.py

admin = Blueprint('admin', __name__)
login_manager = LoginManager()
login_manager.setup_app(admin)

@login_manager.user_loader
def load_user(userid):
    return User.query.get(int(userid))

@admin.route('/login', methods=["GET", "POST"])
def login():
    login_form = LoginForm()
    if request.method == 'POST':
    #####user validation####
        login_user(user)
        return redirect('/')
    return render_template('admin/login.html', login_form=login_form)
Run Code Online (Sandbox Code Playgroud)

run.py

app = Flask(__name__)
app.config.from_object(blog_config)
app.register_blueprint(admin)
if __name__ == "__main__":
    app.run(debug=True)
Run Code Online (Sandbox Code Playgroud)

但是当我发布表单并使用login_user(用户)时,发生了错误.

AttributeError: 'Flask' object has no attribute 'login_manager'
Run Code Online (Sandbox Code Playgroud)

然后我尝试在run.py中使用flask-login,它工作正常.

run.py

login_manager = LoginManager()
login_manager.setup_app(admin)

@login_manager.user_loader
def load_user(userid):
    return User.query.get(int(userid))
Run Code Online (Sandbox Code Playgroud)

那么,我想问的是,flask-login不能在Blueprint对象中使用?谢谢!

小智 27

在Blueprint的注册上也可以登录LoginManager.setup_app():

admin = Blueprint('admin', __name__)
login_manager = LoginManager()

@admin.record_once
def on_load(state):
    login_manager.init_app(state.app)
Run Code Online (Sandbox Code Playgroud)

当蓝图首次注册到应用程序时,将运行on_load.


Tig*_*gra 3

这就是为什么它被称为setup_app

只需将初始化移至您的run.py并将应用程序作为参数传递即可,登录本身可以保留在管理蓝图中