ChoiceEnumArrayField

[danni]

Are you interested in a ChoiceEnumArrayField that builds of django.contrib.postgres.ArrayField ?

[mx-moth]

Yes, I would most certainly be interested in that. I've cobbled together one myself, but a proper reusable implementation would be excellent to include. For reference, I made this to use in the Wagtail admin:

class EnumMultipleChoiceField(forms.MultipleChoiceField):
    widget = forms.CheckboxSelectMultiple

    def __init__(self, enum, members=None, **kwargs):
        self.enum = enum
        self.members = set(members or enum)
        super().__init__(
            choices=[(m.name, str(m)) for m in self.members],
            **kwargs)

    def prepare_value(self, value):
        """
        Convert from a list of Enum members to a list of strings for the widget
        """
        if not value:
            return []
        return [v.name for v in value]

    def to_python(self, value):
        """
        Convert from a list of strings from the widget to a list of Enum members
        """
        items = super().to_python(value)
        members = [self.enum[name] for name in items]
        return sorted(set(members), key=lambda x: x.value)

    def valid_value(self, value):
        return value in self.members

class SomeEnum(ChoiceEnum):
    foo = "Foo"
    bar = "Bar"

class SomePageForm(WagtailAdminPageForm):
    states = EnumMultipleChoiceField(SomeEnum, required=False)

class SomePage(Page):    
    some_choices = ArrayField(EnumChoiceField(SomeEnum), blank=True)
    base_form_class = SomePageForm

[danni]

That looks suspiciously similar to mine :-P

[mx-moth]

Good to know I was on the right track then!