Блог

Я внедряю flask-login на своем веб-сайте (размещенном на Heroku с базой данных PostgreSQL), где в настоящее время я тестирую функцию регистрации. Когда я отправляю форму для регистрации нового пользователя, форма будет отправлена, но база данных не будет обновляться, а сайт не будет перенаправляться.

Вот код для /register маршрута в routes.py :

 @application.route('/register', methods=['GET', 'POST'])
def register():
    if current_user.is_authenticated:
        return redirect(url_for('index'))
    form = RegistrationForm()
    if form.validate_on_submit():
        session = get_session() <--- prints "Session started"
        # Don't need to give the id because it is set to auto-increment (serial type) in PostgreSQL
        user = User(username=form.username.data)
        print("after username")
        user.set_password(form.password.data)
        print("after password")
        session.add(user)
        print("after add")
        session.commit()
        # print("after commit") <--- commenting out this line breaks the program
        return redirect(url_for('login'))
    return render_template('register.html', title='Register', form=form)
 

Форма написана с помощью flask-wtforms. Вот get_session() функция, которая создает сеанс SQLAlchemy, и функция, которая закрывает его в конце каждого запроса (механизм определен в другом месте кода).

 def get_session():
    if 'session' not in g:
        g.session = Session(engine)
        print('Session started')
    return g.session

@application.teardown_request
def teardown_session(exception):
    session = g.pop('session', None)
    if session is not None:
        session.close()
        print('Session torn down')
 

Наконец, вот User класс models.py . Он основан на мега-уроке Мигеля Гринберга по flask.

 from app import login
from flask_login import UserMixin
from sqlalchemy import Column, Integer, Text
from sqlalchemy.ext.declarative import declarative_base
from werkzeug.security import generate_password_hash, check_password_hash

# Declare base to create a table as mapped subclass of the base
base = declarative_base()

# The user inherits the functions of base and UserMixin
class User(base, UserMixin):
    # Map the class to the PostgreSQL table 'users'
    __tablename__ = 'users'
    id = Column('flask-id', Integer, primary_key=True)
    username = Column('username', Text, nullable=False)
    password_hash = Column('password-hash', Text, nullable=False)

    # __repr__ function describes the object
    def __repr__(self):
        return '<User {}>'.format(self.username)

    # Create functions for saving and storing hashed passwords using werkzeug
    # Assigns the hashed password to the User object (the object can be later committed to the database)
    def set_password(self, password):
        self.password_hash = generate_password_hash(password)
    
    # Checks the given plaintext password against the hashed one loaded from the User object
    def check_password(self, password):
        return check_password_hash(self.password_hash, password)

# Flask-login requires a user_loader route to load user objects based on their flask id
# Flask stores the key in memory as a string, so it must be converted to int
@login.user_loader
def load_user(id):
    # Trying to avoid circular dependencies
    from app.routes import get_session
    session = get_session()
    return session.query(User).get(int(id))
 

И вот вывод на консоль при запуске программы на Heroku:

 app[web.1]: Session started
app[web.1]: Session torn down
 

Поскольку печатаются «Сеанс запущен» и «Сеанс сорван», я знаю, что форма может быть проверена и получен запрос POST. Кажется, что после этого программа как бы «выпадает» из функции. Вместо того, чтобы выполнять и в конечном итоге перенаправлять на /login страницу, он просто обновляет /register страницу. Однако, если я просто добавляю комментарии или инструкции печати, они не выпадают таким образом, что заставляет меня думать, что это проблема времени, когда дополнительное время, необходимое для выполнения инструкций печати / комментариев, позволяет базе данных вносить изменения. Однако Python работает синхронно, поэтому я не уверен, что это основная проблема. Буду признателен за любую помощь!