support formatting of enum choices

This commit is contained in:
Frens Jan Rumph 2017-07-07 21:50:37 +02:00
parent 0588f89b89
commit 14a5d62efd

View File

@ -1,3 +1,5 @@
import enum
from django.db import models from django.db import models
from django.utils.encoding import force_text from django.utils.encoding import force_text
@ -17,6 +19,8 @@ singledispatch = import_single_dispatch()
def convert_choice_name(name): def convert_choice_name(name):
if isinstance(name, enum.Enum):
name = name.name
name = to_const(force_text(name)) name = to_const(force_text(name))
try: try:
assert_valid_name(name) assert_valid_name(name)