抽象枚举值的python类型注释

Question

我有一个枚举

from enum import Enum

class StrangeEventTypeVariants(Enum):
    crew_training = "Crew training"
    crew_travel = "Crew travel"
    flight = "Flight"
    maintenance = "Maintenance"
    parking = "Parking"

以及处理抽象枚举的类的函数

class EnumSuggest:
    def __init__(self, enum: Enum):
        self.variants = [{"id": member.name, "name": member.value} for member in enum]

    def get_suggest(self, _):
        return self.variants

但是上面的注释是不正确的，因为

suggest = EnumSuggest(StrangeEventTypeVariants)

我接到下一个警告

Expected type Enum, got Type[StrangeEventTypeVariants]

如何修复它

Answer 1

您没有将Enum的实例传递给EnumSuggest.__init__；您正在传递Enum的子类。类型提示应该反映这一点

class EnumSuggest:
    def __init__(self, enum: Type[Enum]):
        self.variants = [{"id": member.name, "name": member.value} for member in enum]

    def get_suggest(self, _):
        return self.variants