Exception: No user_loader has been installed for this LoginManager

For apps with multiple User models, is there any way to alter the user_loader / load_user() to be able to distinguish between the models? The standard is

@login_manager.user_loader
def load_user(user_id):
    return User.get(user_id)

I need to somehow discern which model the user_id passed belongs to and insert a condition to replace User.get() with the appropriate model.

[alanhamlett]

@davidism could you help release v0.4.2? Would you like me to submit a PR to update the changelog?

[BZ3R0]

You must define the decorator at the same file you instantiated the login_manager. Assuming it’s on yourapp.py you shoud do like below:

login_manager = LoginManager(app)

@login_manager.user_loader
def load_user(id):
    return User.query.filter_by(user_id=id).first()

After that you can delete the @login_manager decorator and its respect function from your models.

Obs. This works on version 0.4.1 of Flask-Login