使用 Model.objects.create() 时未调用 Django 模型 save() 方法

Question

我的模型Image 中有一个主键 ID CharField，我想为新创建的对象创建唯一的 ID。我尝试通过覆盖模型的保存方法来实现这一点：

    def save(self, *args, **kwargs):
        if not self.pk: # is created
            self.id = uuid.uuid4().hex
            while Image.objects.filter(id=self.id).exists():
                self.id = uuid.uuid4().hex
        return super().save(*args,**kwargs)

问题是，save() 在我用Image.objects.create() 创建对象时似乎没有被调用，它只有在我用image=Image(...) 创建对象然后调用image.save() 时才被调用。结果，除非指定，否则新创建的对象没有分配 id，这导致 PostgreSQL 抛出 non_unique primary key 错误。

如何确保在调用 Image.objects.create() 时创建唯一 ID？

Django 版本：1.11.3

更新：我意识到被覆盖的 save() 方法也没有被调用。原来问题是我在迁移中调用模型的save 方法。正如this post 中指出的那样，自定义模型方法在迁移中不可用。我必须将模型的保存方法复制到迁移文件中。

Answer 1

一般情况下无法做到这一点。在您的 if 语句检查 ID 尚不存在和您设置它之间，其他东西可以添加一个具有该 ID 的新行。这就是使用其他解决方案的原因——数据库确保唯一的自动递增 ID，或者唯一机会非常小的 UUID。

幸运的是，您使用了其中一个。使用 UUID 的习惯是假设它们是唯一的。

方法是将返回唯一 ID 的函数设置为字段的默认值：

def uuid_hex():
    return uuid.uuid4().hex

class YourModel(models.Model):
    id = CharField(unique=True, primary_key=True, default=uuid_hex, null=False)

Answer 2

您应该在覆盖模型save 方法时调用真正的save 方法：

def save(self, *args, **kwargs):
    if not self.pk: # is created
        self.id = uuid.uuid4().hex
        while Image.objects.filter(id=self.id).exists():
            self.id = uuid.uuid4().hex
    super(Image, self).save(*args, **kwargs)

检查文档： Overriding predefined model methods

另外，覆盖默认的id 也不是一个好主意。如果您需要另一个唯一字段用作 ID 之类的东西，那么我建议在默认的 id 字段旁边添加另一个字段。