如何将PostgreSQL&39;s"插入.与flask_sqlalChemy一起使用冲突(UPSERT)功能?

问题描述

INSERT语句中的PostgreSQLON CONFLICT子句提供"upsert"功能(即更新现有记录,或者如果不存在这样的记录,则插入新记录)。SQLAlChemy通过PostgreSQL方言的Insert对象上的on_conflict_do_nothingon_conflict_do_update方法(如here所述)支持此功能:

from sqlalchemy.dialects.postgresql import insert

insert_stmt = insert(my_table).values(
    id='some_existing_id',
    data='inserted value'
)

do_nothing_stmt = insert_stmt.on_conflict_do_nothing(
    index_elements=['id']
)

conn.execute(do_nothing_stmt)

do_update_stmt = insert_stmt.on_conflict_do_update(
    constraint='pk_my_table',
    set_=dict(data='updated value')
)

conn.execute(do_update_stmt)
我使用的是flask_sqlalchemy,它为您管理SQLAlChemy的引擎、会话和连接。要向数据库添加元素,我创建模型的一个实例,将其添加到数据库会话,然后调用COMMIT,如下所示:

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
db = SQLAlchemy(app)

class MyTable(db.Model):
    id = db.Column(UUID, primary_key=True)
    data = db.Column(db.String)

relation = MyTable(id=1, data='foo')
db.session.add(relation)
db.session.commit()

因此Insert对象被flask_sqlalchemy完全包裹和遮挡。

如何访问特定于PostgreSQL的方言方法来执行upsert?我是否需要绕过flask_sqlalchemy并创建我自己的会话?如果我这样做,如何确保不会发生冲突?


解决方案

原来您可以在db.session上执行较低级别的语句。因此,解决方案如下所示:

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.dialects.postgresql import insert as pg_insert

app = Flask(__name__)
db = SQLAlchemy(app)

class MyTable(db.Model):
    id = db.Column(UUID, primary_key=True)
    data = db.Column(db.String)

    def __init__(self, _id, *args, **kwargs):
        self.id = _id
        self.data = kwargs['data']

    def as_dict(self):
        return {'id': self.id, 'data': self.data}

    def props_dict(self):
        d = self.as_dict()
        d.pop('id')
        return d

relation = MyTable(id=1, data='foo')
statement = pg_insert(MyTable).
    values(**relation.as_dict()).
    on_conflict_do_update(constraint='id',
                          set_=relation.props_dict())

db.session.execute(statement)
db.session.commit()

我的模型类中的as_dict()props_dict()方法允许我使用构造函数从传入的HTTP请求中过滤移出不需要的属性。

相关文章