from enum import Enum

from image_prediction.formatter.formatters.key_formatter import KeyFormatter
from image_prediction.utils import get_logger

logger = get_logger()


class EnumFormatter(KeyFormatter):
    def format_key(self, key):
        return key.value if isinstance(key, Enum) else key

    def transform(self, obj):
        logger.debug("EnumFormatter.transform")
        raise NotImplementedError