使用 python sqlalchemy 使用 WITH 语句执行原始查询答案

【问题标题】：Using python sqlalchemy to execute raw queries with WITH statement使用 python sqlalchemy 使用 WITH 语句执行原始查询
【发布时间】：2020-04-29 04:21:27
【问题描述】：

我正在尝试使用原始 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 执行查询有何不同？

【问题讨论】：

相关：stackoverflow.com/questions/45347565/…

标签： python postgresql sqlalchemy

【解决方案1】：

我的问题得到了answered on github。

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

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

【讨论】：

考虑关闭已回答的问题 :-) 感谢您的报告。

【解决方案2】：

我遇到了类似的问题，下面的代码帮助了我。如果没有 conn.execute("COMMIT;") 部分，我无法看到我的所有更改都反映在我的数据库中。

with self.engine.begin() as conn:
    conn.execute(sql_query)
    conn.execute("COMMIT;")

【讨论】：