带有注释的Django更新查询集

Question

我想通过使用带注释的值来更新queryset中的所有行。

我有一个简单的模型：

class Relation(models.Model): 
    rating = models.IntegerField(default=0) 

class SignRelation(models.Model): 
    relation = models.ForeignKey(Relation, related_name='sign_relations') 
    rating = models.IntegerField(default=0) 

我想awoid此代码：

for relation in Relation.objects.annotate(total_rating=Sum('sign_relations__rating')): 
    relation.rating = relation.total_rating or 0 
    relation.save() 

，并通过使用像这样做一个SQL请求更新：

Relation.objects.update(rating=Sum('sign_relations__rating')) 

不工作：

TypeError: int() argument must be a string or a number, not 'Sum' 

或

Relation.objects.annotate(total_rating=Sum('sign_relations__rating')).update(rating=F('total_rating')) 

也不起作用：

DatabaseError: missing FROM-clause entry for table "relations_signrelation" 
LINE 1: UPDATE "relations_relation" SET "rating" = SUM("relations_si... 

是否有可能使用Django的ORM为了这个目的？没有关于在文档中使用更新（）和批注（）的信息。

Answer 1

你真的不能这样做。看看the code为update和follow it through一些精美的阅读。

老实说，在Manager定义中放置类似这样的东西有什么问题？把你不想看到的3条线放到经理面前，根据需要给经理打电话。此外，你做的“魔术”要少得多，当下一个开发人员查看你的代码时，他们将不必诉诸几个WTF .. :)

另外，我很好奇，它看起来像你可以使用SQL Join with UPDATE statements，但它是一些经典的SQL hackery ..所以如果你这么倾向，你可以使用Djangos的原始SQL功能;）

Answer 2

UPDATE声明不支持GROUP BY。见例如PostgreSQL Docs，SQLite Docs。

您需要成才这样的：

UPDATE relation 
SET rating = (SELECT SUM(rating) 
       FROM sign_relation 
       WHERE relation_id = relation.id) 

相当于DjangoORM：

from django.db.models.expressions import RawSQL 

Relation.objects.all(). \ 
    update(rating=RawSQL('SELECT SUM(rating) FROM signrelation WHERE relation_id = relation.id', [])) 

或：

from django.db.models import F, Sum 
from django.db.models.expressions import RawSQL 

Relation.objects.all(). \ 
    update(rating=RawSQL(SignRelation.objects. \ 
         extra(where=['relation_id = relation.id']). \ 
         values('relation'). \ 
         annotate(sum_rating=Sum('rating')). \ 
         values('sum_rating').query, [])) 

Answer 3

如果你想避免多次调用数据库，你应该使用transaction.atomic。

了解更多关于Django文档：https://docs.djangoproject.com/en/1.9/topics/db/transactions/#controlling-transactions-explicitly

Answer 4

您可以定义自己的自定义对象经理：

class RelationManager(models.Manager): 
    def annotated(self,*args,*kwargs): 
     queryset = super(RelationManager,self).get_queryset() 
     for obj in queryset: 
       obj.rating = ... do something ... 
     return queryset 

class Relations(models.Model): 
    rating = models.IntegerField(default=0) 
    rating_objects = RelationManager() 

然后在你的代码：

q = Realation.rating_objects.annotated() 

添加ARGS/kwargs定制什么这位经理返回。

Answer 5

解决方法Postgres的：

with connection.cursor() as cursor: 
    sql, params = qs.query.sql_with_params() 
    cursor.execute(""" 
     WITH qs AS ({}) 
     UPDATE foo SET bar = qs.bar 
     FROM qs WHERE qs.id = foo.id 
    """.format(sql), params)