PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

Como usar o recurso INSERT...ON CONFLICT (UPSERT) do PostgreSQL com flask_sqlalchemy?


Acontece que você pode executar uma instrução de nível inferior no db.session . Então, uma solução se parece com isso:
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()

O as_dict() e props_dict() métodos em minha classe de modelo me permitem usar o construtor para filtrar propriedades indesejadas da solicitação HTTP recebida.