python - 如何使枚举指向一个类

Question

假设我有这个枚举来存储所有不同的实体类型

class Entities(Enum):
    TREE = auto()
    ROCK = auto()
    FLOWER = auto()

我想创建一个函数，它采用这些(TREE、ROCK...)枚举之一，并且知道一个枚举对应于我拥有的一个类。例如:

def myFunc(EntityType):
    return type(EntityType)

print(myFunc(Entities.ROCK))
>>>ROCK (where ROCK is an instance of the ROCK class)

如果有办法做到这一点，有没有一种方法，甚至可以初始化类例如:

def myFunc(EntityType):
    myObj = EntityType(pos=(0,0))
    return myObj

Answer 1

如果您放弃 auto 并使用类本身作为 Entities 的值会怎么样？假设 Tree、Rock 和 Flower 是您的类的名称:

class Entities(Enum):
    TREE = Tree
    ROCK = Rock
    FLOWER = Flower

这里的 Entities.TREE.value 是 Tree 的类构造函数。