如何将PostgreSQL&39；s&quot；插入.与flask_sqlalChemy一起使用冲突(UPSERT)功能？

问题描述

INSERT语句中的PostgreSQLON CONFLICT子句提供"upsert"功能(即更新现有记录，或者如果不存在这样的记录，则插入新记录)。SQLAlChemy通过PostgreSQL方言的Insert对象上的on_conflict_do_nothing和on_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请求中过滤移出不需要的属性。