帶有註釋的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)