Django 在多对一关系中引用特定对象

Question

假设我有以下模型：

from django.db import models

class Reporter(models.Model):
    first_name = models.CharField(max_length=30)
    last_name = models.CharField(max_length=30)
    email = models.EmailField()

class Article(models.Model):
    headline = models.CharField(max_length=100)
    pub_date = models.DateField()
    reporter = models.ForeignKey(Reporter, related_name="articles", on_delete=models.CASCADE)

我想向我的Reporter 模型添加一个favorite_article 字段，该字段将引用来自reporter.articles 的特定Article。

一种选择是将信息放入Article 模型中：

class Article(models.Model):
    headline = models.CharField(max_length=100)
    pub_date = models.DateField()
    reporter = models.ForeignKey(Reporter, related_name="articles", on_delete=models.CASCADE)
    is_favorite = models.BooleanField()

但这似乎不是一个非常干净的解决方案。有没有更好的方法来做到这一点？

Answer 1

您建议的方法可行，但在目前的形式下，它允许多篇文章成为一名记者的最爱。通过一些额外的处理，您可以确保每个记者只有一篇（最多）一篇文章是最喜欢的。

对问题Unique BooleanField value in Django? 的几个答案进行一些修改，我们可以限制每个记者一个True 值，而不是整个文章模型的一个True 值。方法是检查同一 Reporter 的其他收藏文章，并在保存实例时将它们设置为不收藏（而不是使用验证限制）。

我还建议在 save 方法中使用单个事务，这样如果保存实例失败，则不会修改其他实例。

这是一个例子：

from django.db import transaction

class Article(models.Model):
    headline = models.CharField(max_length=100)
    pub_date = models.DateField()
    reporter = models.ForeignKey(Reporter, related_name="articles", on_delete=models.CASCADE)
    is_favorite = models.BooleanField(default=False)

    def save(self, *args, **kwargs):
        with transaction.atomic():
            if self.is_favorite:
                reporter_id = self.reporter.id if self.reporter is not None else self.reporter_id
                other_favorites = Article.objects.filter(is_favorite=True, reporter_id=reporter_id)
                if self.pk is not None: # is None when creating a new instance
                    other_favorites.exclude(pk=self.pk)
                other_favorites.update(is_favorite=False)
            return super().save(*args, **kwargs)

为了以防万一，我还更改了使用filter 而不是get 的方法。

然后要获得喜欢的文章为reporter，您可以使用：

try:
    favorite_article = reporter.articles.get(is_favorite=True)
except Article.DoesNotExist:
    favorite_article = None

您可以将其包装到 Reporter 类的方法/属性中。