迭代一个类型而不实例化它

Question

问题

我希望能够迭代一个类型而不实例化它，类似于枚举。

class Foo:
    """Class I want to iterate over without instantiating."""
    ALLOWED_VALUES = (1, 2, 3)

# I want iterating over `Foo` to be equivalent to iterating over `Foo.ALLOWED_VALUES`
for val_from_tuple, val_from_foo in zip(Foo.ALLOWED_VALUES, Foo):
    assert val_from_tuple == val_from_foo

这种行为可以通过枚举实现，但前提是 ALLOWED_VALUES 是有效的 Python 名称。我想在没有这个限制的情况下拥有相同的迭代行为。

我尝试了什么

我尝试在Foo 上将__iter__() 实现为staticmethod，这样就不需要Foo 的实例来获取Iterator。这允许我迭代 Foo.__iter__()，但 iter(Foo) 会引发错误。这似乎是因为iter(Foo) 在type 上寻找__iter__ 方法，而不是在Foo 上（因为Foo 是type 对象）。

class Foo:
    """Class I want to iterate over without instantiating."""
    ALLOWED_VALUES = (1, 2, 3)

    @staticmethod
    def __iter__():
        return Foo.ALLOWED_VALUES

# This works, but isn't what I want because it involves calling `__iter__()` explicitly.
for val in Foo.__iter__():
    print(val)

# This raises an error:
# `TypeError: 'type' object is not iterable`
for val in Foo:
    print(val)

Answer 1

Enum 是可迭代的，因为它使用不同的元类（EnumMeta 而不是type）来创建它。您可以定义自己的元类来提供__iter__ 的定义，而type 本身没有。

class IterableClass(type):
    def __iter__(self):
        yield from self.ALLOWED_VALUES

class Foo(metaclass=IterableClass):
    ALLOWED_VALUES = (1,2,3)

for x, y in zip(Foo.ALLOWED_VALUES, Foo):
    assert x == y