python - 如何从 db.engine.connect().execute 调用中获取 inserted_primary_key

Question

我正在使用:

CPython 2.7.3,
Flask==0.10.1
Flask-SQLAlchemy==0.16
psycopg2==2.5.1
and
postgresql-9.2

尝试通过炼金术从插入调用中获取 PK。

像这样获取引擎:

app = Flask(__name__)
app.config.from_envvar('SOME_VAR')
app.wsgi_app = ProxyFix(app.wsgi_app)  # Fix for old proxyes

db = SQLAlchemy(app)

并在应用程序中执行插入查询:

    from sqlalchemy import text, exc
    def query():
        return db.engine.connect().execute(text('''
        insert into test...'''), kw)
    rv = query()

但是尝试访问 inserted_primary_key 属性，得到:

InvalidRequestError: Statement is not an insert() expression construct.

如何在我的案例中启用 implicit_returning，阅读文档没有帮助？

Answer 1

您可以使用 RETURNING 子句并自行处理:

INSERT INTO test (...) VALUES (...) RETURNING id

然后您可以像通常从查询中检索值一样检索 ID。

请注意，这适用于 Postgres，但不适用于其他数据库引擎，如 MySQL 或 sqlite。

我认为在不使用 ORM 功能的情况下，在 SQLAlchemy 中没有与数据库无关的方法来执行此操作。