ChoiceEnum backport (simplified) for Django < 3.0

django_choice_enum.py

from enum import EnumMeta, Enum

__all__ = ['ChoiceEnum']


class ChoiceEnumMeta(EnumMeta):
    def __iter__(cls):
        return ((tag.value, tag.name) for tag in super().__iter__())

    @property
    def choices(cls):
        return tuple(iter(cls))


class ChoiceEnum(Enum, metaclass=ChoiceEnumMeta):
    """Enum class which can be used in Django model field `choices=` :
    >>> class SomeType(int, ChoiceEnum):
    >>>     TypeA = 1
    >>>     TypeB = 2
    >>>
    >>> from django.db import models
    >>> class MyModel(models.Model):
    >>>     type = models.IntegerField(choices=SomeType.choices)
    >>>
    >>> m = MyModel(type=SomeType.TypeA)
    >>> m.save()
    >>> assert m.type == SomeType.TypeA
    """