PostgreSQL
 sql >> Datenbank >  >> RDS >> PostgreSQL

Wie verwende ich die PostgreSQL-Funktion INSERT...ON CONFLICT (UPSERT) mit Flask_sqlalchemy?

Es stellt sich heraus, dass Sie eine untergeordnete Anweisung auf db.session ausführen können . Eine Lösung sieht also etwa so aus:

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

Das as_dict() und props_dict() Methoden in meiner Modellklasse ermöglichen es mir, den Konstruktor zu verwenden, um unerwünschte Eigenschaften aus der eingehenden HTTP-Anforderung herauszufiltern.