Create an abstract Enum class

后端 未结 3 620
挽巷
挽巷 2020-12-11 02:13

I\'m trying to create an abstract enum (Flag actually) with an abstract method. My final goal is to be able to create a string representation of compound enums,

3条回答
  •  萌比男神i
    2020-12-11 02:51

    Here's how to adapt the accepted answer to the question Abstract Enum Class using ABCMeta and EnumMeta to create the kind of abstract Enum class you want:

    from abc import abstractmethod, ABC, ABCMeta
    from enum import auto, Flag, EnumMeta
    
    
    class ABCEnumMeta(ABCMeta, EnumMeta):
    
        def __new__(mcls, *args, **kw):
            abstract_enum_cls = super().__new__(mcls, *args, **kw)
            try:  # Handle existence of undefined abstract methods.
                absmethods = list(abstract_enum_cls.__abstractmethods__)
                absmethods_str = ', '.join(f'{method!r}' for method in absmethods)
                plural = 's' if len(absmethods) > 1 else ''
                raise TypeError(
                    f"cannot instantiate abstract class {abstract_enum_cls.__name__!r}"
                    f" with abstract method{plural} {absmethods_str}")
            except AttributeError:
                pass
            return abstract_enum_cls
    
    
    class TranslateableFlag(Flag, metaclass=ABCEnumMeta):
    
        @classmethod
        @abstractmethod
        def base(cls):
            pass
    
        def translate(self):
            base = self.base()
            if self in base:
                return base[self]
            else:
                ret = []
                for basic in base:
                    if basic in self:
                        ret.append(base[basic])
                return " | ".join(ret)
    
    
    class Students1(TranslateableFlag):
        ALICE = auto()
        BOB = auto()
        CHARLIE = auto()
        ALL = ALICE | BOB | CHARLIE
    
        @classmethod
        def base(cls):
            return {Students1.ALICE: "Alice", Students1.BOB: "Bob",
                    Students1.CHARLIE: "Charlie"}
    
    
    class Students2(TranslateableFlag):
        ALICE = auto()
        BOB = auto()
        CHARLIE = auto()
        ALL = ALICE | BOB | CHARLIE
    
    # Abstract method not defined - should raise TypeError.
    #    @classmethod
    #    def base(cls):
    #        ...
    

    Result:

    Traceback (most recent call last):
      ...
    TypeError: cannot instantiate abstract class 'TranslateableFlag' with abstract method 'base'
    

提交回复
热议问题