SQLAlchemy typed IntegerEnum

integerenum.py

from typing import Type, TypeVar
from enum import IntEnum
import sqlalchemy

E = TypeVar('E', bound=IntEnum)


class IntegerEnum(sqlalchemy.types.TypeDecorator):
    impl = sqlalchemy.Integer

    def __init__(self, enum_type: Type[E], *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._enum_type = enum_type

    def process_bind_param(self, value: E, dialect: str) -> int:
        return value.value

    def process_result_value(self, value: int, dialect: str) -> E:
        return self._enum_type(value)