Limit choices of an enum about django-enumfields HOT 2 OPEN

enum_field_with_limited_choices = EnumField(
    EnumWithManyChoices,
    max_length=100,
    choices=[
        (EnumWithManyChoices.Choice1, EnumWithManyChoices.Choice1.label),
        (EnumWithManyChoices.Choice2, EnumWithManyChoices.Choice2.label),
    ],
)

class EnumField(ChoiceField):
    def __init__(self, enum, lenient=False, ints_as_names=False, **kwargs):
        """
        :param enum: The enumeration class.
        :param lenient: Whether to allow lenient parsing (case-insensitive, by value or name)
        :type lenient: bool
        :param ints_as_names: Whether to serialize integer-valued enums by their name, not the integer value
        :type ints_as_names: bool
        """
        self.enum = enum
        self.lenient = lenient
        self.ints_as_names = ints_as_names
        # the if-block is new
        if 'choices' not in kwargs:
            kwargs['choices'] = tuple((e.value, getattr(e, 'label', e.name)) for e in self.enum)
        super().__init__(**kwargs)

class EnumSupportSerializerMixin:
    enumfield_options = {}
    enumfield_classes_to_replace = (ChoiceField, CharField, IntegerField)

    def build_standard_field(self, field_name, model_field):
        field_class, field_kwargs = (
            super().build_standard_field(field_name, model_field)
        )
        if isinstance(model_field, EnumFieldMixin) and field_class in self.enumfield_classes_to_replace:
            field_class = EnumSerializerField
            field_kwargs['enum'] = model_field.enum
            field_kwargs['choices'] = [(enum.value, label) for enum, label in model_field.choices]
            field_kwargs.update(self.enumfield_options)
        return field_class, field_kwargs