Flask-Login: войти в систему и переадресовать на домашнюю страницу после регистрации

Question

Использование входа в колбу, попытка входа в систему пользователя после регистрации пользователем. user_login() работает для моего логина, однако после регистрации он перенаправляется на маршрут входа, а не на главную страницу. Поэтому мой вопрос: может ли login_user() работать только на маршруте «входа»?

Вот основной вариант того, что я сейчас:

class User(flask_login.UserMixin): 
def __init__(self, userName, email, firstName): 
     self.userName = userName 
     self.id = email 
     self.email = email 
     self.firstName = firstName 

@login_manager.user_loader 
def user_loader(email): 
    """ code to get userdata from database """ 

    user = User(userName,email,firstName) 

    return user 

@app.route('/register', methods=['GET','POST']) 
def register(): 
    if request.method == 'GET': 
     return render_template('register.html') 
    elif request.method == 'POST': 
     """ code for creating user from form data """ 
     #create session for user 
     user = User(userName,email,firstName) 

     flask_login.login_user(user) 

     return redirect(url_for('home')) 

@app.route('/login', methods=['GET','POST']) 
def login(): 

    if request.method == 'GET': 
     return render_template('login.html') 

    elif request.method == 'POST': 
     """ code for authorizing a user """ 

     user = User(userName,email,firstName) 

     flask_login.login_user(user) 
     return flask.redirect(flask.url_for('home')) 
    else: 
     return render_template('login.html') 

@app.route('/home') 
@flask_login.login_required 
def home(): 
    return render_template('home.html') 

UPDATE: я не нагружал мой пользователь правильно в моем user_loader

Answer 1

После регистрации пользователя, вы перенаправлении прямо на маршрут /home, который требует, чтобы пользователь был зарегистрирован. Но вы на самом деле не зарегистрировали их на этом этапе, а только создали пользователя.

Вы должны добавить следующее в вашем /register маршрута после того, как пользователь был создан:

flask_login.login_user(user)