使用session.query通过SQLAlchemy ORM更新连接表

时间:2015-10-15 16:30:18

标签: python mysql sqlalchemy

使用MySQL,我想生成这个SQL:

array(0, 1, 2, 3, 4... 555)

这是我的SQLAlchemy查询:

UPDATE tableA
INNER JOIN tableB
ON tableA.some_id = tableB.some_id
SET tableA.foo = 1
WHERE tableB.bar IN ('baz','baaz')

但它生成的SQL是这样的(多表更新,没有连接条件,这不是我想要的):

session.query(tableA).join(tableB, tableA.some_id == tableB.some_id) \
                     .filter(tableB.bar.in_(['baz','baaz']))\
                     .update({tableA.foo: 1})

我尝试将.join更改为另一个.filter来指定连接条件,但没有解决问题。 如何强制使用这个简单的更新语句来进行正确的连接?

2 个答案:

答案 0 :(得分:7)

从0.7.4版开始,sqlalchemy.sql.expression.update允许您引用WHERE子句中的多个表。有了这个,您可以构建并执行如下表达式:

users.update().values(name='ed').where(
        users.c.name==select([addresses.c.email_address]).\
                    where(addresses.c.user_id==users.c.id).\
                    as_scalar()
        )

(直接来自上面的链接)

ValAyal遇到的问题实际上是因为Query.join()不支持Query.update()。不幸的是,直到0.9.1,它才会默默地生成类似上面分享的ValAyal之类的查询。 changelog notes for 0.9.1注意到行为被修改为发出警告:

  

[orm] [bug]查询不支持连接,子选择或特殊FROM   使用Query.update()或Query.delete()方法时的子句;   如果像Query.join()这样的方法,而不是默默地忽略这些字段   或者调用了Query.select_from(),发出警告。作为   1.0.0b5这会引发错误。

     

参考文献:#3349

我们实际上遇到了这个我今晚工作的地方,并发现我们的代码实际上发出了以下警告(表示1.0中的错误):

SAWarning: Can't call Query.update() or Query.delete() when join(), outerjoin(), select_from(), or from_self() has been called.  This will be an exception in 1.0
  self._validate_query_state()

在我们的案例中,我们选择将更新转换为对一个表的选择和更新。

答案 1 :(得分:1)

我认为我有完全相同的问题。这是我的解决方案:

query = update(Model).values(field=123)
query = query.where(Model.parent_model_id == ParentModel.id)
query = query.where(ParentModel.grand_parent_id == GrandParentModel.id)
query = query.where(GrandParentModel.name == 'foobar')
session.execute(query)