PostgreSQL
 sql >> Database >  >> RDS >> PostgreSQL

Come utilizzare la funzione INSERT...ON CONFLICT (UPSERT) di PostgreSQL con flask_sqlalchemy?

Si scopre che puoi eseguire un'istruzione di livello inferiore su db.session . Quindi una soluzione assomiglia a questa:

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

Il as_dict() e props_dict() i metodi nella mia classe del modello mi consentono di utilizzare il costruttore per filtrare le proprietà indesiderate dalla richiesta HTTP in entrata.