nice choices class for

choices.py

"""
Example usage:

# declare
class Color(Choices):
    RED = 'Red'
    GREEN = 'Green'
    BLUE = 'Blue'
   
# use on model
class Row(Model):
    color = models.TextField(choices=Color, default=Color.RED)

# or on form
class RowForm(Form):
    color = forms.ChoiceField(choices=Color)
"""


from enum import Enum, EnumMeta


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

    def __call__(self, *args, **kwargs):
        """Create or lookup value.

        If no arguments are passed (django.forms.ChoiceField), return iterator
        to choices.
        """
        if args or kwargs:
            return super().__call__(*args, **kwargs)
        return self.__iter__()


class Choices(Enum, metaclass=ChoicesMeta):
    pass