python - Django/MySQL 唯一约束如何将 NULL 视为相等

Question

我有一个 Django 模型，它有一个递归字段。简化版本如下。这个想法大致是在sql中有一个树形数据结构。显然，我遇到的问题是 Django does not treat NULLs as equal 。现在的问题是，由于每棵树的根都必然有一个“空指针”，我可以有两个相同的树，但 Django 会因为 NULL 值而将它们视为不同的。如何实现下面的 UniqueConstraint，以便两个具有 NULL 链接值和相等节点值的“链接”对象将被视为相同，并且无法通过 UniqueConstraint 测试？谢谢。

class Link(models.Model):
    node = models.ForeignKey(Node, on_delete=models.CASCADE) 

    link = models.ForeignKey('self', on_delete = models.CASCADE, null=True)


    class Meta:

        constraints = [
                models.UniqueConstraint(['node', 'link'], name='pipe_unique')
                ]

编辑 当然，理想情况下，约束将由数据库强制执行。但即使我可以通过 Hook 某处或使用自定义约束在应用程序逻辑中强制执行它，那也足够了。

Answer 1

您可以使用custom constraint来做到这一点

UniqueConstraint(fields=['node'], condition=Q(link__isnull=True), name='unique_root_node')

编辑:

如果您希望手动添加检查，您可以在 Link 的 save 方法以及 clean 方法中执行此操作，以便在您保存实例之前，它就可以在任何模型表单中运行

def clean(self):
    if self.node_id and not self.link_id:
        if self.__class__.objects.exclude(pk=self.pk).filter(node=self.node, link__isnull=True).exists():
            raise ValidationError(f'A root node already exists for {self.node}')

排除pk=self.pk可以避免更新对象时与自身发生冲突

def save(self, *args, **kwargs):
    self.clean()
    super().save(*args, **kwargs)