From 14a5d62efda46b6ea6c71c9fcbf9d06099e2dcf9 Mon Sep 17 00:00:00 2001 From: Frens Jan Rumph Date: Fri, 7 Jul 2017 21:50:37 +0200 Subject: [PATCH] support formatting of enum choices --- graphene_django/converter.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/graphene_django/converter.py b/graphene_django/converter.py index b1a8837..839ecc3 100644 --- a/graphene_django/converter.py +++ b/graphene_django/converter.py @@ -1,3 +1,5 @@ +import enum + from django.db import models from django.utils.encoding import force_text @@ -17,6 +19,8 @@ singledispatch = import_single_dispatch() def convert_choice_name(name): + if isinstance(name, enum.Enum): + name = name.name name = to_const(force_text(name)) try: assert_valid_name(name)