python - SQLAlchemy 多列相关更新

Question

我已经能够使用文档中显示的相关更新构造 here更新表中的一列。

例如:

sel = select([UpdateTable.column]).\
      where(UpdateTable.id == OrigTable.id)

up = update(OrigTable).values(column=sel) 

生成的 SQL 如下:

UPDATE origtable SET column=(SELECT updatetable.column
FROM updatetable
WHERE updatetable.id = origtable.id)

是否可以使用 Declaritive 或 Query Api 来更新一个选择的多个列？

我正在尝试在 PostgreSQL 中模拟如下内容:

UPDATE origtable
   SET
     column1 = updatetable.column1,
     column2 = updatetable.column2
   FROM updatetable 
   WHERE origtable.id == updatetable.id

编辑:

目前的 SQLAlchemy api 似乎无法将此格式化为单个语句。

但是，使用来自 here 的解决方案似乎可以使用两个选择的解决方法.

Answer 1

因为这是针对 Postgresql 的，我们可以使用 SQLAlchemy 的 multiple table update支持。该文档仅适用于核心，但 Query API构建在 Core 之上，我们可以 apply that knowledge :

session.query(OrigTable).\
    filter(OrigTable.id == UpdateTable.id).\
    update({OrigTable.column1: UpdateTable.column1,
            OrigTable.column2: UpdateTable.column2},
           synchronize_session=False)

结果

UPDATE origtable
SET column1=updatetable.column1,
    column2=updatetable.column2
FROM updatetable
WHERE origtable.id = updatetable.id