refactor how _convert_choices_to_enum is evaluated.

This commit is contained in:
a-c-sreedhar-reddy 2020-03-04 10:18:43 +05:30
parent aeb04d5b5c
commit 040a48b1f3

View File

@ -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