Question

我正在使用：

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

尝试使用alchemy从插入调用中获取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属性，获取：

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中有一种与数据库无关的方法。

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

如何从db.engine.connect（）获取inserted_primary_key。执行调用

3 个答案: