时间戳

首页 > 解决方案 > Django / MySQL唯一约束如何将NULL视为平等

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

问题描述

我有一个具有递归字段的 Django 模型。简化版本如下。这个想法大致是在sql中有一个树形数据结构。我遇到的问题是，显然，Django 不将 NULLs 视为 equal。现在的问题是，由于每棵树的根都必须有一个“空指针”，我可以有两棵相同的树，但由于 NULL 值，Django 会将它们视为不同的。如何实现下面的 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')
                ]

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

标签： pythonmysqldjangomodel

解决方案

您可以使用自定义约束来执行此操作

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

编辑：

如果您希望手动添加检查，您可以在save方法Link和方法中执行此操作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)

推荐阅读