python - 使用 python sqlalchemy 通过WITH语句执行原始查询

Question

我正在尝试使用原始 sqlalchemy text() 查询将值插入 Postgres11 数据库。当我通过 psql-client 运行以下 SQL 查询时，它可以正常工作:

WITH a AS (
  INSERT INTO person (id) 
       VALUES ('a')
    RETURNING id
) 
INSERT INTO person_info (person_id) 
     SELECT id 
       FROM a;

所有行均已正确插入:

# select id from person;
 id 
----
 a
(1 row)

但是，如果我在 sqlalchemy 中创建一个引擎并使用相同的查询对其调用执行，它会成功运行，但不会插入任何行:

>>> engine.execute("WITH a AS (INSERT INTO person (id) VALUES ('b') RETURNING id) INSERT INTO person_info (person_id) SELECT id from a")
<sqlalchemy.engine.result.ResultProxy object at 0x7f25e6c2a090>

但没有插入新行:

# select id from person;
 id 
----
 a
(1 row)

通过 psql-client 运行查询与通过 sqlalchemy 执行查询有何不同？

Answer 1

我的问题得到了answered on github .

解决方案是将执行包装在事务上下文中:

with engine.begin() as conn:
   conn.execute("whatever")