python - sqlalchemy插入查询不返回结果

Question

我尝试从插入查询中选择返回值:

HOST_INSERT_QUERY = (
    Host.__table__.insert()
        .values(
            name=bindparam("name"),
            source=bindparam("source"),
            type=bindparam("type")
                )
        .returning(Host.__table__.c.id)
)
result = db_conn.execute(HOST_INSERT_QUERY, values)

我从 db.echo 看到了什么:

info sqlalchemy.engine.base.Engine INSERT INTO hosts (name, source, type) VALUES ( %(name)s, %(source)s, %(type)s) RETURNING hosts.id
info sqlalchemy.engine.base.Engine ({ 'name': 'hhh', 'source': '["import"]', 'type': 'host'},{...})
debug:  Added 2 hosts

确实有效。主机插入数据库，但我在调试器中看到一些问题；结果(ResultProxy)有:

rowcount = 2 #right!
is_insert = True #right!
return_rows=False #WHYYYY????????

如果我尝试:

result.fetсhall()

AttributeError: 'ResultProxy' object has no attribute 'fetсhall'

版本:

• SQLAlchemy 1.3.1
• psycopg-二进制 2.7.7

注意! 此代码仅在插入单个记录时有效。插入多行时，无法从RETURNING中拉出值。

Answer 1

You cannot get results from executemany ，如果您传递一组多于 1 行的数据来插入，则 SQLAlchemy 在下面使用它。这就是您观察 return_rows=False 的原因。一种解决方法是使用一个多 VALUES 子句“内联”行，而不是使用 bindparams 和 executemany:

HOST_INSERT_QUERY = (
    Host.__table__.insert()
        .values(values)
        .returning(Host.__table__.c.id)
)

result = db_conn.execute(HOST_INSERT_QUERY)

这作为单个“多值”语句执行，因此避免使用 executemany，并可能返回结果。请注意，返回的 ID 的顺序可能与插入的行的顺序不匹配。