如何简化Python类实例的迭代？实现直接对类进行成员检查

为什么你的代码报错？ #

你用@classmethod定义的__iter__不会被Python识别为类本身的迭代方法——因为Player类本身是type的实例，Python在尝试迭代类时，只会去它的元类（也就是type）里找__iter__方法，而不是类自身的类方法。所以当执行'P1' in Player时，找不到合法的迭代器，就抛出了TypeError。

两种解决方案 #

方案1：直接实现__contains__类方法（推荐） #

因为in操作符优先调用__contains__方法来判断成员关系，相比实现迭代器，这种方式更直接高效，完全匹配你的需求：

class Player():
    all = []
    def __init__(self, name):
        self.name = name
        Player.all.append(name)

    @classmethod
    def __contains__(cls, item):
        # 直接委托给all列表的成员检查
        return item in cls.all

测试代码：

p1 = Player('P1')
print('P1' in Player.all)  # 输出 True
print('P1' in Player)     # 输出 True

方案2：让类本身可迭代（支持遍历+in操作） #

如果后续需要支持for name in Player这样的遍历操作，可以通过元类来实现类的迭代能力：

# 定义一个元类，实现__iter__方法
class IterableClass(type):
    def __iter__(cls):
        # 返回类属性all的迭代器
        return iter(cls.all)

# 让Player类继承这个元类
class Player(metaclass=IterableClass):
    all = []
    def __init__(self, name):
        self.name = name
        Player.all.append(name)

测试代码：

p1 = Player('P1')
p2 = Player('P2')
print('P1' in Player)  # 输出 True
# 支持遍历类本身
for name in Player:
    print(name)  # 依次输出 P1、P2

额外优化：如果all存储的是实例而非名称 #

如果你的all应该存储Player实例（更符合类设计逻辑），可以修改__contains__来同时支持实例和名称检查：

class Player():
    all = []
    def __init__(self, name):
        self.name = name
        Player.all.append(self)

    @classmethod
    def __contains__(cls, item):
        if isinstance(item, str):
            # 检查名称是否存在于实例的name属性中
            return any(player.name == item for player in cls.all)
        # 检查是否是实例本身
        return item in cls.all

测试代码：

p1 = Player('P1')
print(p1 in Player)    # 输出 True
print('P1' in Player)  # 输出 True

内容的提问来源于stack exchange，提问作者user20325621