您如何在SQLAlchemy ORM中表达多表更新（UPDATE FROM）？

Question

CREATE TABLE foo (
name text NOT NULL,
deleted_at timestamp without time zone
);

CREATE TABLE bar (
    name text NOT NULL,
    status_id int
);

UPDATE bar set status_id=1
FROM foo
WHERE status_id <> 1 AND foo.name = bar.name AND foo.deleted_at is null;

当我尝试使用ORM执行此操作时，我最终遇到此错误

InvalidRequestError: Can't call Query.update() or Query.delete() when join(), outerjoin(), select_from(), or from_self() has been called

我想使用ORM，以便在更新命令完成之前使用更改来更新会话。

Answer 1

使用ORM发行多表更新的方法如下：

session.query(Bar).\
    filter(Bar.status_id != 1,
           Bar.name == Foo.name,
           Foo.deleted_at.is_(None)).\
    update({Bar.status_id: 1}, synchronize_session=False)

Query
API
文档中的脚注，用于指向Core多表更新文档的更新链接：

update()通过在WHERE子句中指定多个表，SQLAlchemy构造隐式支持这两种模式

它也扩展到了ORM Query API，这并不奇怪，因为ORM是建立在Core之上的。结果查询为：

UPDATE bar SET status_id=%(status_id)s
FROM foo
WHERE bar.status_id != %(status_id_1)s
  AND bar.name = foo.name
  AND foo.deleted_at IS NULL

从您的错误中我怀疑您有一些类似的东西

session.query(Bar).select_from(Foo)...update(...)

由于错误状态而被接受。您必须在WHERE子句中（例如filter()在查询API中）隐式传递FROM表。

达到

我想使用ORM，以便在更新命令完成之前使用更改来更新会话。

您必须将相应地更改synchronize_session为'fetch'或'evaluate'。