Django 未强制执行枚举值

发布于09月13日

在Django中，我正在try 创建一个基于枚举的数据库字段.我正在避免使用外部库，并且只想使用普通的Django4.

本质上，我是在复制Python - Covert enum to Django models.CharField choices tuple中给出的例子

但是，枚举值不会强制执行.我可以很容易地创建一个包含animal = 'foo'的数据库条目，它会很高兴地将其写入数据库.

我预计这会产生一个IntegrityError，因为我显式地使用了一个枚举，所以只有有效的值才会在db中结束.

推荐答案

Django确实没有强制枚举值，或者至少不是现成的.严格地说，这could件事是可以完成的:它将 Select 权交给了移民.但迁移系统不会创建特定类型，或者至少在默认情况下不会创建.

但是，您可以使用CheckConstraint ^[Django-doc]来强制执行这些操作:

from django.db.models import Q


class AnimalModel(models.Model):
    class Animal(models.TextChoices):
        DOG = 'dog'
        CAT = 'cat'

    animal = models.CharField(choices=Animal.choices)

    class Meta:
        constraints = [
            models.CheckConstraint(
                check=Q(animal__in=list(Animal)), name='animal_valid'
            )
        ]

话虽如此，通常您会通过表单、序列化程序等来处理数据，以验证这一点.通常，仅仅通过模型输入数据是一个好主意:Django的模型默认情况下出于性能原因跳过了很多验证.