Python：使动态创建的类（如类本身）可迭代？

Question

我从这段代码 sn-p 开始，据我了解，它本质上是一个类工厂，用于模拟其他语言的“枚举”类型：

def enum(*sequential, **named):
    enums = dict(zip(sequential, range(len(sequential))), **named)
    return type('Enum', (), enums)

（我取自这里：How can I represent an 'Enum' in Python?）

我知道它是如何工作的，而且确实如此，但我想让我的动态生成的类（它们的类型为“枚举”）可迭代，以便我可以执行以下操作以进行完整性检查：

MyEnum = enum('FOO', 'BAR', 'JIMMY')
def func(my_enum_value):  # expects one of the MyEnum values
    if not my_enum_value in MyEnum:
        raise SomeSortOfException

但是，为了使健全性检查工作，我需要使 MyEnum 可迭代。我在这里读到：http://pydanny.blogspot.com/2007/10/required-methods-to-make-class-iterable.html 我需要按顺序添加 iter len contains 和 getitem 方法使其可迭代。我开始走这条路（重写枚举代码）但卡住了：

def enum(*sequential, **named):
    enums = dict(zip(sequential, range(len(sequential))), **named)
    enums['_enums'] = enums.copy() # (so I'd have them in the Class in order to use for the methods I'll be implementing to make it iterable)
    Enum = type('Enum', (), enums)

    Enum.__iter__ = #Er, how do I do this? Guess I'll ask SO
    Enum.__len__ =  # etc. 
    # . . . 
    return Enum

那么，如何使生成的 Enum 类可迭代，以便我可以对它们使用 'in' 词？这对我来说不是生死攸关的事，但我通过这样做学习了大量的 Python。

Answer 1

使用比type 更智能的元类。

class EnumMC(type):
  def __contains__(self, val):
    return val in self.__dict__

def enum(*sequential, **named):
  enums = dict(zip(sequential, range(len(sequential))), **named)
  return EnumMC('Enum', (), enums)

MyEnum = enum('FOO', 'BAR', 'JIMMY')
def func(my_enum_value):  # expects one of the MyEnum values
    if not my_enum_value in MyEnum:
        raise ValueError()

func('FOO')
func('QUUX')

您可能希望使用实际属性来存储枚举键，而不是依赖于类字典。

Answer 2

class Enum(dict): pass
def enum(*sequential, **named):
    return Enum(zip(sequential, range(len(sequential))), **named)

这更容易阅读，而且它支持迭代。

编辑：你仍然需要做

MyEnum = enum('FOO', 'BAR', 'JIMMY')

那你就可以了

'FOO' in MyEnum

你也可以使用

class Enum(dict):
    def __init__(self, *sequential, **named):
        self.update(zip(sequential, range(len(sequential))))
        self.update(named)
def __getattr__(self, attr):
    if attr in self:
        return self[attr]
    else:
        return super(Enum, self).__getattr__(attr)

做同样的事情。

编辑 2：另一种选择。

from collections import namedtuple

def enum(*sequential):
    return namedtuple('Enum', sequential)(*sequential)

仍然以同样的方式调用它，你可以使用MyEnum.FOO，它支持in。还编辑了我上面的类以支持属性访问。

如果您想要连续的整数值，您也可以使用range(len(sequential)) 填写namedtuple 值（不是键），或者如果您想要指定任意值，请使用**kwargs 然后kwargs.keys() 和kwargs.values() .

最后一个解决方案交叉发布到How can I represent an 'Enum' in Python?

Answer 3

我不知道为什么enum 需要返回class 而不是instance，这样你就可以轻松实现__contains__，我也不确定为什么我们甚至需要python 中的枚举，反正如果你需要在类级别添加__contains__，你需要一个元类，就在这里

class EnumMeta(type):
    def __contains__(self, x):
        return x in self.__dict__

class EnumBase(object):
    __metaclass__ = EnumMeta

def enum(*sequential, **named):
    enums = dict(zip(sequential, range(len(sequential))), **named)
    return type('Enum', (EnumBase,), enums)

MyEnum = enum('FOO', 'BAR', 'JIMMY')

print 'foobar in MyEnum','foobar' in MyEnum
print 'FOO in MyEnum', 'FOO' in MyEnum

输出：

foobar in MyEnum False
FOO in MyEnum True