django-enumfields2 3.0.2

EnumField, EnumIntegerField

from enumfields import EnumField
from enum import Enum

class Color(Enum):
    RED = 'r'
    GREEN = 'g'
    BLUE = 'b'

class MyModel(models.Model):

    color = EnumField(Color, max_length=1)

其他地方

m = MyModel.objects.filter(color=Color.RED)

EnumIntegerField与EnumField工作方式相同，但底层存储机制是IntegerField而不是CharField。

class MyForm(forms.Form):

    color = EnumField(Color, max_length=1).formfield()

枚举

通常，您只需使用正常的PEP435样式枚举，但是django-enumfields还包括其自己的枚举版本，具有一些额外的功能。特别是，它提供了用于例如在管理下拉列表中使用的标签的智能定义。默认情况下，它将通过将您的常量名称首字母大写来创建标签。您可以使用嵌套的“Labels”类提供自定义标签。

from enumfields import EnumField, Enum  # Our own Enum class

class Color(Enum):
    RED = 'r'
    GREEN = 'g'
    BLUE = 'b'

    class Labels:
        RED = 'A custom label'

class MyModel(models.Model):
    color = EnumField(Color, max_length=1)

assert Color.GREEN.label == 'Green'
assert Color.RED.label == 'A custom label'

EnumFieldListFilter

enumfields.admin.EnumFieldListFilter提供，允许在list_filter中使用枚举。

from enumfields.admin import EnumFieldListFilter

class MyModelAdmin(admin.ModelAdmin):
    list_filter = [('color', EnumFieldListFilter)]

Django Rest Framework集成

EnumSupportSerializerMixin 混合类允许你在 DRF 序列化器中使用枚举。

# models.py
from enumfields import EnumField
from enum import Enum

class Color(Enum):
    RED = 'r'
    GREEN = 'g'
    BLUE = 'b'

class MyModel(models.Model):
    color = EnumField(Color, max_length=1)


# serializers.py
from enumfields.drf.serializers import EnumSupportSerializerMixin
from rest_framework import serializers
from .models import MyModel

class MyModelSerializer(EnumSupportSerializerMixin, serializers.ModelSerializer):
    class Meta:
        model = MyModel
        fields = '__all__'