Truy vấn cập nhật Django với chú thích

Question

Tôi muốn cập nhật tất cả các hàng trong queryset bằng cách sử dụng giá trị chú thích.

Tôi có một mô hình đơn giản:

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) 

Và tôi muốn awoid mã này:

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

Và làm cập nhật trong một SQL-yêu cầu bằng cách sử dụng một cái gì đó như thế này:

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

Không hoạt động:

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

hoặc

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

Cũng không hoạt động:

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

Có thể sử dụng ORM Django cho mục đích này? Không có thông tin về việc sử dụng cập nhật() và chú thích() cùng nhau trong tài liệu.

Answer 1

Bạn thực sự không thể thực hiện việc này. Hãy xem the code cho update và follow it through để đọc tốt.

Thành thật mà nói, có vấn đề gì khi đặt thứ gì đó như thế này trong định nghĩa của Người quản lý? Đặt 3 dòng bạn không muốn đưa vào chế độ xem của bạn vào người quản lý, hãy gọi cho người quản lý đó nếu cần. Ngoài ra, bạn đang thực hiện "ma thuật" ít hơn nhiều và khi nhà phát triển tiếp theo xem mã của bạn, họ sẽ không phải sử dụng một vài WTF .. :)

Ngoài ra, tôi tò mò và có vẻ như bạn có thể sử dụng SQL Join with UPDATE statements nhưng đó là một số hackery SQL cổ điển .. vì vậy, nếu bạn đang nghiêng như vậy, bạn có thể sử dụng chức năng Djangos liệu SQL cho rằng;)

Answer 2

UPDATE tuyên bố không hỗ trợ GROUP BY. Xem ví dụ PostgreSQL Docs, SQLite Docs.

Bạn cần someting như thế này:

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

tương đương trong DjangoORM:

from django.db.models.expressions import RawSQL 

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

hay:

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

Nếu bạn muốn tránh nhiều cuộc gọi đến các cơ sở dữ liệu, bạn nên sử dụng transaction.atomic.

Tìm hiểu thêm về Django tài liệu: https://docs.djangoproject.com/en/1.9/topics/db/transactions/#controlling-transactions-explicitly

Answer 4

Bạn có thể xác định đối tượng tùy chỉnh riêng quản lý của bạn:

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() 

Sau đó, trong mã của bạn:

q = Realation.rating_objects.annotated() 

Thêm args/kwargs tùy chỉnh những gì người quản lý này trả về.

Answer 5

Giải pháp cho 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)