Как переопределить метод save, для изменения другой модели Django?

Question

[gh0sty]

Создаю систему рейтинга на Django.

У меня есть 3 модели - Review (отзывы), Profile (профиль) и Advert (объявления).
Модель Peview содержит integer поле stars и поле ad.
ad - форейнг ки на объект Advert, к которому прикреплен отзыв.
Модель Advert в свою очередь содержит форейн ки на объект Profile, владельца объявления.
Модель Profile содержит integer поле rating.

И я хочу при сохранении (создании или обновлении) объекта модели Review, переопределять rating в модели Profile.
Можно ли сделать такое? Есть вариант получше?
Вот мой код метода save для Review, я пока не понимаю, что делает super, поэтому он нерабочий:

def save(self, *args, **kwargs):
        super().save(*args, **kwargs) # надеюсь эта строчка сохраняет объект review
        all_ads = Advert.objects.filter(profile_id=self.ad.profile.id) # эта строчка берет все объявления пользователя (работает)
        reviews = []
        # цикл выбирает все review, на все объявления этого пользователя (работает)
        for rev in Review.objects.filter(status='active'):
            if rev.ad in all_ads:
                reviews.append(rev)
        rev_stars = 0
        for r in reviews:
            rev_stars += r.stars
        new_rating = round((rev_stars+3)/len(reviews)) # подсчет рейтинга (среднее из stars + округление + default-значение "3") (работает, вроде считает)
        self.ad.profile.rating = new_rating # внесение изменений в модель профиля, Review --> Advert --> Profile --> rating
        # super(self.ad.profile, self).save(*args, **kwargs)
        # не знаю как сохранить обновленную модель

Answer 1

[Сергей Горностаев]

Если ваш сервис сейчас не испытывает лютых нагрузок в сотни запросов в секунду, и вопрос об оптимизации в жёлтой зоне ещё не встал, то лучше не хранить поле rating, а добавлять аннотацией в момент выборки данных из БД. Если же СУБД надо разгрузить любой ценой, то надо не save переопределять, а обработчик сигнала pre_save объявить, в котором записывать это поле.

Comments

[gh0sty]

Окей, можете рассказать подробнее про pre_save, я знаю, что такое post и pre__save, но не до конца понимаю когда происходит само сохранение, какая нужна для этого функция и что вообще делает super. И почему именно pre_save, если данные мы получаем после сохранения?

[Сергей Горностаев]

gh0sty, никакой функции не надо, надо просто записать в поле rating нужное вам значение. super позволяет обращаться к методам суперкласса, в обработчиках сигналов вам это не нужно. Сигнал pre_save вызывается перед сохранением, post_save - после сохранения. Вам нужно изменить поле перед записью в базу, а не после, поэтому и pre_save.

[gh0sty]

Вот только пришел. Я скорее всего не так описал ситуацию (крч надо было использовать post_save, т.к. у меня есть еще поле status у review, которое учитывается в запросе). Спасибки!
Вот код, если кому интересно:

@receiver(post_save, sender=Review)
def rating_update(sender, instance, *args, **kwargs):
    all_ads = Advert.objects.filter(profile_id=instance.ad.profile.id)
    rev_stars = 3
    rev_count = 1
    for rev in Review.objects.filter(status='active'):
        if rev.ad in all_ads:
            rev_count += 1
            rev_stars += rev.stars
    instance.ad.profile.rating = round(rev_stars//rev_count)
    instance.ad.profile.save()