python - 在 Python 中将字符串转换为枚举

Question

将字符串转换为 Enum 子类的相应实例的正确方法是什么？似乎 getattr(YourEnumType, str) 可以完成这项工作，但我不确定它是否足够安全。

举个例子，假设我有一个类似的枚举

class BuildType(Enum):
    debug = 200
    release = 400

给定字符串 'debug'，我怎样才能得到 BuildType.debug 作为结果？

Answer 1

此功能已内置于 Enum :

>>> from enum import Enum
>>> class Build(Enum):
...   debug = 200
...   build = 400
... 
>>> Build['debug']
<Build.debug: 200>

成员名称区分大小写，因此如果正在转换用户输入，您需要确保大小写匹配:

an_enum = input('Which type of build?')
build_type = Build[an_enum.lower()]