Django - 处理“枚举模型”

Django

Django - 处理“枚举模型”

在开发 Web 应用程序时，经常会遇到需要定义一组固定值的场景，例如性别、颜色或状态等。在 Django 中，我们可以使用枚举模型来处理这种情况，它可以帮助我们更好地组织和管理这些固定值。

什么是枚举模型

枚举模型是一种将固定值作为选项的模型。它允许我们在数据库中存储固定值的代码，并使用这些代码来表示实际值。通过使用枚举模型，我们可以避免在多个模型中重复定义相同的固定值，并且可以轻松地更新和维护这些值。

定义枚举模型

在 Django 中，我们可以使用 choices 参数来定义枚举模型。choices 参数接受一个包含元组的列表，每个元组包含两个值：一个是存储在数据库中的代码值，另一个是表示实际值的字符串。

下面是一个使用枚举模型的例子，定义了一个表示性别的模型：

Python
from Django.db import models
class Gender(models.TextChoices):
    MALE = 'M', '男'
    FEMALE = 'F', '女'
    OTHER = 'O', '其他'
class Person(models.Model):
    name = models.CharField(max_length=100)
    gender = models.CharField(
        max_length=1,
        choices=Gender.choices,
        default=Gender.OTHER
    )

在上面的例子中，Gender 是一个继承自 models.TextChoices 的枚举类。我们定义了三个选项：MALE、FEMALE 和 OTHER，并分别指定了它们的代码值和实际值。然后，在 Person 模型中，我们使用 gender 字段来表示性别，并指定了可选的选择项为 Gender.choices。

使用枚举模型

使用枚举模型非常简单。我们可以像使用普通的字段一样使用枚举字段，并且可以通过代码值来获取实际值。

下面是一些使用枚举模型的例子：

Python
# 创建一个人的实例，并设置性别为男
person = Person.objects.create(name='John', gender=Gender.MALE)
# 获取该人的性别
print(person.gender)  # 输出：M
# 获取该人的性别的实际值
print(person.get_gender_display())  # 输出：男

在上面的例子中，我们首先创建了一个名为 John 的人，并将其性别设置为 MALE。然后，我们可以通过访问 gender 字段来获取性别的代码值，并通过调用 get_gender_display() 方法来获取性别的实际值。

枚举模型是 Django 中处理固定值的一种有效方式。通过定义枚举模型，我们可以更好地组织和管理固定值，并且可以轻松地使用它们。在开发 Web 应用程序时，使用枚举模型可以提高代码的可读性和可维护性。

案例代码

Python
from Django.db import models
class Gender(models.TextChoices):
    MALE = 'M', '男'
    FEMALE = 'F', '女'
    OTHER = 'O', '其他'
class Person(models.Model):
    name = models.CharField(max_length=100)
    gender = models.CharField(
        max_length=1,
        choices=Gender.choices,
        default=Gender.OTHER
    )
# 创建一个人的实例，并设置性别为男
person = Person.objects.create(name='John', gender=Gender.MALE)
# 获取该人的性别
print(person.gender)  # 输出：M
# 获取该人的性别的实际值
print(person.get_gender_display())  # 输出：男

参考链接

- Django 文档：Djangoproject.com/">https://docs.Djangoproject.com/

- Django - 处理“枚举模型”：Djangoproject.com/">https://docs.Djangoproject.com/en/3.2/ref/models/fields/#choices