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

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

Answer 1

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

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

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

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

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

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