flask-login не может использоваться в объекте Blueprint?

у меня есть вопрос относительно колбы-логина и чертежа.

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.py

app = Flask(__name__)
app.config.from_object(blog_config)
app.register_blueprint(admin)
if __name__ == "__main__":
    app.run(debug=True)

но когда я отправляю форму и использую login_user (user), произошла ошибка.

AttributeError: 'Flask' object has no attribute 'login_manager'

затем я пытаюсь использовать flask-login in run.py-он отлично работает.

run.py

login_manager = LoginManager()
login_manager.setup_app(admin)

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

Итак, что я хочу спросить, колба-логин не может использоваться в объекте Blueprint? THX!

2 ответов


вот почему он называется setup_app

просто переместите инициализацию в свой run.py и передать приложение в качестве параметра И сам логин может остаться внутри admin blueprint


также возможно LoginManager.setup_app () при регистрации Blueprint:

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

@admin.record_once
def on_load(state):
    login_manager.init_app(state.app)

on_load будет запущен, когда схема будет впервые зарегистрирована в приложении.