Django - save() 更新重复键

Question

我几乎没有允许用户对视频进行评分的应用程序。

用户只能评价一次。 所以我定义了模型的唯一性。

但他应该能够改变他的利率。 因此 save() 应该更新重复的键

class VideoRate(models.Model):
  """Users can Rate each Video on the criterias defined for the topic"""
  user = models.ForeignKey(User)
  video = models.ForeignKey(VideoFile)
  crit = models.ForeignKey(VideoCrit)
  rate = models.DecimalField(max_digits=2, decimal_places=1, choices=RATE_CHOICES)
  class Meta:
    unique_together = (('user', 'video', 'crit'),)
    verbose_name = 'Video Rating'

如果我

rate = VideoRate(user_id=1, video_id=1, crit_id=1, rate=2)
rate.save()

它正在保存评级，但如果我

rate = VideoRate(user_id=1, video_id=1, crit_id=1, rate=3)
rate.save()

得到正常错误

IntegrityError: (1062, "Duplicate entry '1-1-1' for key 'user_id'")

即使我使用 force_update=True （因为仅基于在主键上）

如果评级已经存在，有没有办法更新评级而无需事先检查数据？

Answer 1

要更新现有评级，您实际上必须拥有要更新的评级。如果您知道该对象可能不存在，请使用get_or_create：

rate, created = VideoRate.objects.get_or_create(user_id=1, video_id=1, crit_id=1)
rate.rate = 2
rate.save()

您可以使用update()来缩短该过程：

VideoRate.objects.filter(user_id=1, video_id=1, crit_id=1).update(rate=2)

但如果评级不存在，这将默默失败- 它不会创建一个。

Answer 2

首先，您必须检查评级是否存在。因此，您可以使用 Daniel Roseman 所说的或使用 exists，但您无法通过简单的更新来解决此问题，因为更新不会创建新记录...

rating = 2
rate, created = VideoRate.objects.get_or_create(user_id=1, video_id=1, crit_id=1,
    defaults={'rate':rating})#if create, also save the rate infdormation

if not created:# update
    rate.rate = rating
    rate.save()

您可以使用默认传递exrta参数，因此如果是插入，数据库记录将使用所有必需的信息创建，您不需要再次更新它...

文档

更新：这个答案就像问题一样很旧。正如 @peterthomassen 提到的，Django 现在有 update_or_create()方法

Answer 3

Django 4.1 添加了对此类冲突的支持。在后端，它运行 INSERT...ON DUPLICATE KEY UPDATE 查询。

上述单个查询的示例：

rate = VideoRate(user_id=1, video_id=1, crit_id=1, rate=2)
rate.save()

# updating rate object if it exists, else create one
rate = VideoRate(user_id=1, video_id=1, crit_id=1, rate=3)
VideoRate.objects.bulk_create(
    [rate],
    update_conflicts=True,
    update_fields=['rate']
)