使用Flask-SQLAlchemy获取插入的记录

Question

下面的代码片段演示了一个超级简单的Flask应用程序，如果没有冲突(使用on_conflict_do_nothing方法)，它会在Account表中插入记录。

我希望在响应中以JSON对象的形式返回插入的记录。

有没有一种“内置”的方法来返回记录？我知道我可以获得插入的主键I，但我想要的是整个记录集。

from flask import Flask, request, jsonify
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.dialects.postgresql import insert

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = <postgres>
db = SQLAlchemy(app)


class Account(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    transaction_id = db.Column(db.Integer, unique=True,)
    username = db.Column(db.String(80), unique=True, nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)

    def __repr__(self):
        return '<User %r>' % self.username


@app.route("/", methods=['POST'])
def insert():
    stmt = (
        insert(Account)
        .values(request.json)
        .on_conflict_do_nothing(index_elements=[Account.transaction_id])
    )
    db.session.execute(stmt)
    db.session.commit()

    result = None  # return inserted records

    return jsonify({
        'status': 'success',
        'result': result
    })

Answer 1

正如@snakecharmerb在评论中指出的那样，我使用了returning。

def insert():
    stmt = (
        insert(Account)
        .values(request.json)
        .returning(Account.__table__)
        .on_conflict_do_nothing(index_elements=[Account.transaction_id])
    )
    result_proxy = db.session.execute(stmt)
    db.session.commit()

    result = [dict(row) for row in result_proxy]

    return jsonify({
        'status': 'success',
        'result': result
    })