Django模型中基于Choices的字段约束实现方法咨询

要实现你描述的约束规则，需要同时在应用层和数据库层做验证，既保证用户操作时的明确错误提示，也能防止绕过应用层的直接数据库操作。

一、应用层验证（Model.clean方法） #

在模型中重写clean方法，在保存前检查约束条件，抛出针对性的错误：

from django.core.exceptions import ValidationError
from django.db import models

class ExampleChoices(models.TextChoices):
    A = "A", "a"
    B = "B", "b"
    C = "C", "c"
    D = "D", "c"

class ExampleModel(models.Model, PublishModelMixin):
    property = models.CharField(choices=ExampleChoices.choices, default=ExampleChoices.A)

    def clean(self):
        super().clean()
        current_val = self.property

        # 处理选项C的约束
        if current_val == ExampleChoices.C:
            # 检查C是否已存在（更新时排除自身）
            existing_c = ExampleModel.objects.filter(property=ExampleChoices.C)
            if self.pk:
                existing_c = existing_c.exclude(pk=self.pk)
            if existing_c.exists():
                raise ValidationError("选项C只能存在一条记录")
            
            # 检查系统中是否有D
            if ExampleModel.objects.filter(property=ExampleChoices.D).exists():
                raise ValidationError("系统中已存在选项D，无法使用选项C")
        
        # 处理选项D的约束
        elif current_val == ExampleChoices.D:
            # 检查D是否已存在（更新时排除自身）
            existing_d = ExampleModel.objects.filter(property=ExampleChoices.D)
            if self.pk:
                existing_d = existing_d.exclude(pk=self.pk)
            if existing_d.exists():
                raise ValidationError("选项D只能存在一条记录")
            
            # 检查系统中是否有C
            if ExampleModel.objects.filter(property=ExampleChoices.C).exists():
                raise ValidationError("系统中已存在选项C，无法使用选项D")

    def save(self, *args, **kwargs):
        # 强制触发full_clean（包含clean方法），避免直接调用save时跳过验证
        self.full_clean()
        super().save(*args, **kwargs)

二、数据库层约束 #

仅应用层验证可能被绕过（比如直接通过数据库客户端操作），因此需要添加数据库级别的约束：

class ExampleModel(models.Model, PublishModelMixin):
    property = models.CharField(choices=ExampleChoices.choices, default=ExampleChoices.A)

    # ... 上述clean和save方法 ...

    class Meta:
        constraints = [
            # 确保选项C在表中唯一
            models.UniqueConstraint(
                fields=['property'],
                condition=models.Q(property=ExampleChoices.C),
                name='unique_c_property'
            ),
            # 确保选项D在表中唯一
            models.UniqueConstraint(
                fields=['property'],
                condition=models.Q(property=ExampleChoices.D),
                name='unique_d_property'
            ),
            # 确保C和D不同时存在于表中
            models.CheckConstraint(
                check=~models.Q(property=ExampleChoices.C) | ~models.Q(property=ExampleChoices.D),
                name='c_d_exclusive'
            )
        ]

注意事项 #

• CheckConstraint需要数据库支持：PostgreSQL完全支持，MySQL 8.0.16+支持，SQLite默认禁用CHECK约束，需在项目配置中启用（不推荐生产环境用SQLite）。
• 添加约束后，需生成并执行新的迁移文件，让数据库生效约束。

内容的提问来源于stack exchange，提问作者Xander