ChoiceEnum

choiceenum.py

import enum

class ChoiceEnum(enum.Enum):
    @staticmethod
    def tr(func):
        label_for = classmethod(lambda cls, member: _(func(member.name)))
        return type('ChoiceEnumTr', (), {'label_for': label_for})

    @classmethod
    def label_for(cls, member):
        return _(member.name)

    @classmethod
    def choices(cls, empty_label=None):
        choices = tuple((m.value, cls.label_for(m)) for m in cls)
        if empty_label:
            choices = ((None, empty_label),) + choices
        return choices


class Type(ChoiceEnum.tr(str.title), ChoiceEnum):
    manager = 0
    contact = 1
    # ugettext_noop("Manager")
    # ugettext_noop("Contact")