sql >> Base de Datos >  >> RDS >> PostgreSQL

¿Cómo utilizar la función INSERTAR... EN CONFLICTO (UPSERT) de PostgreSQL con Flask_sqlalchemy?

Resulta que puede ejecutar una declaración de nivel inferior en db.session . Así que una solución se parece a esto:

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

El as_dict() y props_dict() Los métodos en mi clase de modelo me permiten usar el constructor para filtrar las propiedades no deseadas de la solicitud HTTP entrante.