如何从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)

并在app中执行插入查询：

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

但是尝试访问inserted_primary_key属性时，get：

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

如何启用implicit_returning在我的情况下，阅读文档没有帮助？

Answer 1

您可以使用RETURNING子句并自己处理：

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

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

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

我不认为在SQLAlchemy中没有一种不依赖于数据库的方法可以在不使用ORM功能的情况下做到这一点。

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