Peewee EnumField (custom field type based on IntegerField)

enumfield.py

import enum
from abc import ABC
from peewee import IntegerField


class EnumField(IntegerField, ABC):
    """
    This class enable an Enum like field for Peewee
    """

    def __init__(self, enum: Type[enum.Enum], *args, **kwargs):
        super(IntegerField, self).__init__(*args, **kwargs)
        self.enum = enum

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

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