From 040a48b1f38e844956700788407cd4fdbadd8699 Mon Sep 17 00:00:00 2001 From: a-c-sreedhar-reddy Date: Wed, 4 Mar 2020 10:18:43 +0530 Subject: [PATCH] refactor how _convert_choices_to_enum is evaluated. --- graphene_django/types.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/graphene_django/types.py b/graphene_django/types.py index 0c0cb1c..967d8fb 100644 --- a/graphene_django/types.py +++ b/graphene_django/types.py @@ -45,14 +45,10 @@ def construct_fields( # in there. Or when we exclude this field in exclude_fields. # Or when there is no back reference. continue - _convert_choices_to_enum = convert_choices_to_enum if not isinstance(_convert_choices_to_enum, bool): # then `convert_choices_to_enum` is a list of field names to convert - if name in _convert_choices_to_enum: - _convert_choices_to_enum = True - else: - _convert_choices_to_enum = False + _convert_choices_to_enum = name in _convert_choices_to_enum converted = convert_django_field_with_choices( field, registry, convert_choices_to_enum=_convert_choices_to_enum