我正在与 Python 枚举作斗争。我创建了一个包含各种字段的枚举类:
class Animal(Enum):
DOG = "doggy"
CAT = "cute cat"
我知道我可以通过值访问这个枚举,即通过传递
Animal("doggy")
我将拥有 Animal.DOG
。然而,我想实现相同的目标,但相反,假设我有 "DOG"
,我用它来查找 "doggy"
。
示例:
str_to_animal_enum("DOG") = Animal.DOG
我想我可以重写 Enum 类的 init 函数来检查我们作为参数放入的字符串名称是否在 Enum 名称中具有代表,如果是,那么我们有一个匹配。
像这样:
def __init__(cls, name):
for enum in all_enums:
if enum.name == name:
return enum
但是这个解决方案看起来计算量很大。假设我有 100 个名称要查找,对于每个名称,我们必须在最坏的情况下进行 N 次迭代,其中 N 是枚举类中的枚举数。
我认为应该有更好的东西,但我只是不知道是什么。您对如何实施有什么想法吗?谢谢:)
按照手册:
>>> Animal.DOG.value
'doggy'
>>> Animal.DOG.name
'DOG'
>>> # For more programmatic access, with just the enum member name as a string:
>>> Animal['DOG'].value
'doggy'