- en
- Language: ru
- Documentation version: 2.0
4. Как обновить денормализованные поля в других моделях при сохранении?
У вас есть такие модели.
class Category(models.Model):
name = models.CharField(max_length=100)
hero_count = models.PositiveIntegerField()
villain_count = models.PositiveIntegerField()
class Meta:
verbose_name_plural = "Categories"
class Hero(models.Model):
name = models.CharField(max_length=100)
category = models.ForeignKey(Category, on_delete=models.CASCADE)
# ...
class Villain(models.Model):
name = models.CharField(max_length=100)
category = models.ForeignKey(Category, on_delete=models.CASCADE)
# ...
Вам нужно, чтобы hero_count и villain_count обновлялись при создании новых объектов.
Вы можете сделать что-то вроде этого
class Hero(models.Model):
# ...
def save(self, *args, **kwargs):
if not self.pk:
Category.objects.filter(pk=self.category_id).update(hero_count=F('hero_count')+1)
super().save(*args, **kwargs)
class Villain(models.Model):
# ...
def save(self, *args, **kwargs):
if not self.pk:
Category.objects.filter(pk=self.category_id).update(villain_count=F('villain_count')+1)
super().save(*args, **kwargs)
Обратите внимание, что мы не использовали self.category.hero_count += 1, так как update будет выполнять обновление БД.
Альтернативным методом является использование signals. Вы можете сделать это следующим образом.
from django.db.models.signals import pre_save
from django.dispatch import receiver
@receiver(pre_save, sender=Hero, dispatch_uid="update_hero_count")
def update_hero_count(sender, **kwargs):
hero = kwargs['instance']
if hero.pk:
Category.objects.filter(pk=hero.category_id).update(hero_count=F('hero_count')+1)
@receiver(pre_save, sender=Villain, dispatch_uid="update_villain_count")
def update_villain_count(sender, **kwargs):
villain = kwargs['instance']
if villain.pk:
Category.objects.filter(pk=villain.category_id).update(villain_count=F('villain_count')+1)
4.1. Сигналы против переопределения .save
Поскольку любой из сигналов .save может быть использован для поведения сохранения, когда следует использовать какой из них? Я следую простому правилу.
Если ваши поля зависят от модели, которую вы контролируете, переопределите
.save.Если ваши поля зависят от модели стороннего приложения, которое вы не контролируете, используйте сигналы.