Как обновить значение набора запросов в Django? - PullRequest
Новая
       15

Как обновить значение набора запросов в Django?

2 голосов
/ 28 мая 2019

Я написал скрипт Python в моем проекте.Я хочу обновить значение поля.

Вот мои режимы 


class News_Channel(models.Model):
    name = models.TextField(blank=False)
    info = models.TextField(blank=False)
    image = models.FileField()
    website = models.TextField()
    total_star = models.PositiveIntegerField(default=0)
    total_user = models.IntegerField()

    class Meta:
        ordering = ["-id"]

    def __str__(self):
        return self.name

class Count(models.Model):
    userId = models.ForeignKey(User, on_delete=models.CASCADE)
    channelId = models.ForeignKey(News_Channel, on_delete=models.CASCADE)
    rate = models.PositiveIntegerField(default=0)

    def __str__(self):
        return self.channelId.name

    class Meta:
        ordering = ["-id"]

Это мой скрипт на python: 

from feed.models import Count, News_Channel


def run():
    for i in range(1, 11):
        news_channel = Count.objects.filter(channelId=i)
        total_rate = 0
        for rate in news_channel:
            total_rate += rate.rate
        print(total_rate)
        object = News_Channel.objects.filter(id=i)
        print(total_rate)
        print("before",object[0].total_star,total_rate)
        object[0].total_star = total_rate
        print("after", object[0].total_star)
        object.update()

После подсчета total_rate изТаблица количества Я хочу обновить общее значение звезды в таблице News_Channel.Я не могу сделать это и получить данные до обновления и после обновления как ноль.Хотя total_rate имеет значение.

Ответы [ 3 ]

4 голосов
/ 28 мая 2019

Проблема

Причина, по которой это не удается, заключается в том, что здесь object - это QuerySet из News_Channel с, да, QuerySet может содержать ровно один News_Channel, но это не имеет значения.

Если вы затем используете object[0], вы делаете запрос в базу данных, чтобы получить первый элемент и десериализовать его в объект News_Channel.Затем вы устанавливаете total_star этого объекта, но вы никогда не сохраняете этот объект.Вы набираете .update() только для всего набора запросов, что приводит к другому независимому запросу.

Это можно исправить с помощью: 

objects = News_Channel.objects.filter(id=i)
<b>object = objects[0]</b>
object.total_star = total_rate
object<b>.save()</b>

Или, если вам не нужна проверка, вы можете повысить производительность с помощью: 

News_Channel.objects.filter(id=i).update(total_star=total_rate)

Обновление все News_Channel с

Если вы хотите обновить все News_Channel с, вам на самом деле лучше использовать Subquery здесь: 

from django.db.models import OuterRef, Sum, Subquery

subq = Subquery(
    Count.objects.filter(
        channelId=OuterRef('id')
    ).annotate(
        total_rate=Sum('rate')
    ).order_by('channelId').values('total_rate')[:1]
)

News_Channel.objects.update(total_star=subq)
3 голосов
/ 28 мая 2019

Причина в том, что object в вашем случае является набором запросов, и после попытки обновления object[0] вы не сохраняете результаты в БД и не обновляете набор запросов.Чтобы заставить его работать, вы должны передать поле, которое вы хотите обновить, в метод обновления.

Итак, попробуйте это: 

def run():
    for i in range(1, 11):
        news_channel = Count.objects.filter(channelId=i)
        total_rate = 0
        for rate in news_channel:
            total_rate += rate.rate
        print(total_rate)
        object = News_Channel.objects.filter(id=i)
        print(total_rate)
        print("before",object[0].total_star,total_rate)
        object.update(total_star=total_rate)
        print("after", object[0].total_star)
2 голосов
/ 28 мая 2019

News_Channel.total_star можно рассчитать с помощью агрегации 

news_channel_obj.count_set.aggregate(total_star=Sum('rate'))['total_star']

Затем вы можете использовать это в своем скрипте: 

object.total_star = object.count_set.aggregate(total_star=Sum('rate'))['total_star']

Или, если вы делаетенет необходимости кэшировать это значение, поскольку производительность не является проблемой, вы можете удалить поле total_star и добавить его в качестве свойства в News_Channel модель 

@property
def total_star(self):
    return self.count_set.aggregate(total_star=Sum('rate'))['total_star']
...