第七节 可迭代对象

from collections.abc import Iterable
from collections.abc import Iterator

# 判断是否是可迭代对象
print(isinstance([11, 22, 33], Iterable))

class Classmate(object):
    def __init__(self):
        self.name = list()

    def add(self, name):
        self.name.append(name)

    def __iter__(self):
        """要使一个对象变成可迭代对象，必须调用这个方法"""
        return ClassIterator(self)  #将ClassIterator类实例化的一个对象当作返回值


class ClassIterator():

    def __init__(self, obj):
        self.obj = obj.name
        self.num = 0
    def __iter__(self):
        pass

    def __next__(self):
        if self.num <= len(self.obj):  #设置if条件，让原来的for不会一直重复取同一个值
            ret = self.obj[self.num]
            self.num += 1
            return ret
        else:
            raise StopIteration  #抛出异常，for取数,如果没有这句话，for取完数后，会一直取__next__(self)返回的none

classmate = Classmate()
classmate.add("111")
classmate.add("222")
classmate.add("333")

print(isinstance(classmate, Iterable))  # 判断是否是可迭代对象
classmate_iterator = iter(classmate)  #iter函数会自动调用classmate中的__iter__方法，取得其返回值，如果返回对象中有__next__(self)方法，则返回该对象的地址，否则报错
print(isinstance(classmate_iterator, Iterator))  #判断是否是迭代器
for x in classmate:
    print(x)

'''
最终版，将两个类合并
'''
class Classmate(object):
    def __init__(self):
        self.name = list()
        self.num = 0
    def add(self, name):
        self.name.append(name)

    def __iter__(self):
        """要使一个对象变成可迭代对象，必须调用这个方法"""
        return self  # 这里将自己的引用作为返回值，让人for循环会自动调用iter()函数，从这个引用里找到__next__(self)方法，执行
    def __next__(self):
        if self.num <= len(self.name ):  # 设置if条件，让原来的for不会一直重复取同一个值
            ret = self.name [self.num]
            self.num += 1
            return ret
        else:
            raise StopIteration  # 抛出异常，for取数,如果没有这句话，for取完数后，会一直取__next__(self)返回的none