PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Comment utiliser la fonctionnalité INSERT...ON CONFLICT (UPSERT) de PostgreSQL avec flask_sqlalchemy ?

Il s'avère que vous pouvez exécuter une instruction de niveau inférieur sur le db.session . Donc, une solution ressemble à ceci :

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()

Le as_dict() et props_dict() Les méthodes de ma classe de modèle me permettent d'utiliser le constructeur pour filtrer les propriétés indésirables de la requête HTTP entrante.