Enum field support

I’ve had a look around but can’t seem to find any reference to Peewee supporting enum fields, other than a short mailing list thread with no outcome, and a user contributed recipe for Postgres which is untested. There are also arguments for not using enum.

All the supported Peewee databases have support for enum fields, and a quick look at the source indicates it would be trivial to add it in.

Is there any reason why enum fields have not been added yet? Would you consider accepting a PR for this?

[bratao]

This code works good for Python 3.4+ enum for MySQL

class enum_field(CharField):
    """
    This class enable a Enum like field for Peewee
    """
    def __init__(self, choices, *args, **kwargs):
        self.choices = choices
        super(CharField, self).__init__(*args, **kwargs)

    def db_value(self, value):
        return value.name

    def python_value(self, value):
        return self.choices(value)

class my_table(Model):
    class StatusChoices(Enum):
        WAITING = 1
        READY_TO_SEND = 2
        UPLOADING = 3
        OCR = 4
        DONE = 5

    status = enum_like_field(null=False, choices=StatusChoices)

[leinardi]

Yet another version of the EnumField. This version stores the enum value as CharField but restores the correct Enum type when reading it:

class EnumField(CharField):
    """
    This class enable an Enum like field for Peewee
    """

    def __init__(self, choices: Callable, *args: Any, **kwargs: Any) -> None:
        super(CharField, self).__init__(*args, **kwargs)
        self.choices = choices
        self.max_length = 255

    def db_value(self, value: Any) -> Any:
        return value.value

    def python_value(self, value: Any) -> Any:
        return self.choices(type(list(self.choices)[0].value)(value))