__new__
您的类型上的方法用于enum.Enum
创建枚举值的新实例,因此Types.Unknown
,Types.Source
等单例实例。枚举调用(例如Types('nl')
由 处理EnumMeta.__call__
,您可以对其进行子类化。
使用名称别名适合您的用例
在这种情况下,压倒__call__
一切可能是矫枉过正。相反,您可以轻松地使用名称别名:
class Types(enum.Enum):
Unknown = 0
Source = 1
src = 1
NetList = 2
nl = 2
这Types.nl
是一个别名,将返回与相同的对象Types.Netlist
。然后您按名称访问成员(使用Types[..]
索引访问);所以Types['nl']
有效并返回Types.Netlist
。
您关于无法免费迭代枚举值别名的断言是不正确的。迭代明确不包括别名:
遍历枚举成员不提供别名
别名是Enum.__members__
有序字典的一部分,如果您仍然需要访问这些。
一个演示:
>>> import enum
>>> class Types(enum.Enum):
... Unknown = 0
... Source = 1
... src = 1
... NetList = 2
... nl = 2
... def __str__(self):
... if self is Types.Unknown: return '??'
... if self is Types.Source: return 'src'
... if self is Types.Netlist: return 'nl'
...
>>> list(Types)
[<Types.Unknown: 0>, <Types.Source: 1>, <Types.NetList: 2>]
>>> list(Types.__members__)
['Unknown', 'Source', 'src', 'NetList', 'nl']
>>> Types.Source
<Types.Source: 1>
>>> str(Types.Source)
'src'
>>> Types.src
<Types.Source: 1>
>>> str(Types.src)
'src'
>>> Types['src']
<Types.Source: 1>
>>> Types.Source is Types.src
True
这里唯一缺少的是将未知模式转换为Types.Unknown
; 我会为此使用异常处理:
try:
scheme = Types[scheme]
except KeyError:
scheme = Types.Unknown
覆盖__call__
如果您想将您的字符串视为值,并使用调用而不是项目访问,这就是您覆盖__call__
元类方法的方式:
class TypesEnumMeta(enum.EnumMeta):
def __call__(cls, value, *args, **kw):
if isinstance(value, str):
# map strings to enum values, defaults to Unknown
value = {'nl': 2, 'src': 1}.get(value, 0)
return super().__call__(value, *args, **kw)
class Types(enum.Enum, metaclass=TypesEnumMeta):
Unknown = 0
Source = 1
NetList = 2
演示:
>>> class TypesEnumMeta(enum.EnumMeta):
... def __call__(cls, value, *args, **kw):
... if isinstance(value, str):
... value = {'nl': 2, 'src': 1}.get(value, 0)
... return super().__call__(value, *args, **kw)
...
>>> class Types(enum.Enum, metaclass=TypesEnumMeta):
... Unknown = 0
... Source = 1
... NetList = 2
...
>>> Types('nl')
<Types.NetList: 2>
>>> Types('?????')
<Types.Unknown: 0>
请注意,我们在这里将字符串值转换为整数,并将其余部分留给原始 Enum 逻辑。
完全支持值别名
因此,enum.Enum
支持名称别名,您似乎需要值别名。覆盖__call__
可以提供一种复制,但是我们可以通过将值别名的定义放入枚举类本身来做得更好。例如,如果指定重复名称会给您提供值别名怎么办?
您必须提供一个子类,enum._EnumDict
因为它是防止名称被重用的类。我们假设第一个枚举值是默认值:
class ValueAliasEnumDict(enum._EnumDict):
def __init__(self):
super().__init__()
self._value_aliases = {}
def __setitem__(self, key, value):
if key in self:
# register a value alias
self._value_aliases[value] = self[key]
else:
super().__setitem__(key, value)
class ValueAliasEnumMeta(enum.EnumMeta):
@classmethod
def __prepare__(metacls, cls, bases):
return ValueAliasEnumDict()
def __new__(metacls, cls, bases, classdict):
enum_class = super().__new__(metacls, cls, bases, classdict)
enum_class._value_aliases_ = classdict._value_aliases
return enum_class
def __call__(cls, value, *args, **kw):
if value not in cls. _value2member_map_:
value = cls._value_aliases_.get(value, next(iter(Types)).value)
return super().__call__(value, *args, **kw)
然后,您可以在枚举类中定义别名和默认值:
class Types(enum.Enum, metaclass=ValueAliasEnumMeta):
Unknown = 0
Source = 1
Source = 'src'
NetList = 2
NetList = 'nl'
演示:
>>> class Types(enum.Enum, metaclass=ValueAliasEnumMeta):
... Unknown = 0
... Source = 1
... Source = 'src'
... NetList = 2
... NetList = 'nl'
...
>>> Types.Source
<Types.Source: 1>
>>> Types('src')
<Types.Source: 1>
>>> Types('?????')
<Types.Unknown: 0>