Flask-SQLAlchemy: как условно вставить или обновить строку

В моем приложении используется комбинация Flask, Flask-SQLAlchemy, Flask-WTF и Jinja2.

В своем текущем воплощении у меня есть таблица настроек. В таблице будет только одна запись с одним полем. Первоначально таблица содержит нулевые записи.

Чего я хочу достичь:

Учитывая, что в db нет записей, тогда покажите пустую форму, готовую для ввода пользователем
Учитывая, что запись существует, покажите запись и
если пользователь изменит значение, а затем обновит rec в db.

Вот мой код:

models.py

class Provider(db.Model):
    id = db.Column(db.Integer, primary_key = True)
    rssfeed = db.Column(db.String(120), unique = True)

    def __init__(self, rssfeed):
        self.rssfeed = rssfeed

    def __repr__(self):
        return '<NZBMatrix feed url %r>' % self.rssfeed

forms.py

class SettingsForm(Form):
    rssfeed = TextField('rssfed', validators= [Required()])

views.py

    @app.route('/settings', methods=["GET","POST"])
    def settings():
    """ show settings """
        provider = Provider.query.get(1)
        form = SettingsForm(obj=provider)
        print provider

        if request.method == "POST" and form.validate():
            if Provider.query.get(1) is None:
                provider = Provider(rssfeed=form.rssfeed.data)
                form.populate_obj(provider)
                db.session.add(provider)
                db.session.commit()
                flash("Settings added")

        return render_template("settings.html", form=form)

В этом случае этот код создает запись, если она не существует, но столбец rssfeed пуст.

Как я могу изменить этот код так, чтобы он был INSERT, если запись не существует, и UPDATE, если это так?

Ответы

Ответ 1

Как только ваша форма будет подтверждена и т.д.,

Чтобы добавить новую запись:

new_provider = Provider(form.rssfeed.data)
db.session.add(new_provider)
db.session.commit()

Чтобы обновить существующую запись:

existing_provider = Provider.query.get(1) # or whatever
# update the rssfeed column
existing_provider.rssfeed = form.rssfeed.data
db.session.commit()

Фокус в обновлении заключается в том, что вам просто нужно изменить конкретное поле и выполнить фиксацию. отдых выполняется по сеансу db. Я думаю, вы используете функцию слияния, которая теперь устарела в SQLAlchemy.

Ответ 2

Мне удалось решить проблему с этими изменениями в файле view.py:

@app.route('/settings', methods=["GET","POST"])
def settings():
    """ show settings """
    provider = Provider.query.get(1)
    form = SettingsForm(request.form,obj=provider)

    if request.method == "POST" and form.validate():
        if provider:
            provider.rssfeed = form.rssfeed.data
            db.session.merge(provider)
            db.session.commit()
            flash("Settings changed")
            return redirect(url_for("index"))
        else:
            provider = Provider(form.rssfeed.data)
            db.session.add(provider)
            db.session.commit()
            flash("Settings added")
            return redirect(url_for("index"))
    return render_template("settings.html", form=form)