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 中没有与 db 无关的方法来执行此操作。

Answer 2

您是否有任何理由进行文本查询而不是普通的 sqlalchemy insert()？如果您使用的是 sqlalchemy，将查询改写为：

from sqlalchemy import text, exc, insert

# in values you can put dictionary of keyvalue pairs
# key is the name of the column, value the value to insert
con = db.engine.connect()
ins = tablename.insert().values(users="frank")
res = con.execute(ins)
res.inserted_primary_key
[1] 

这样 sqlalchemy 将为您完成绑定。

Answer 3

你可以使用 lastrowid

rv = query()
rv.lastrowid