6.6协程-迭代器

 迭代器一（迭代器与初始化分拆两个方法）

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

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

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

    def __iter__(self):
        '''如果想要一个对象称为一个可以迭代的对象，即可以使用for，那么必须实现__iter__方法'''
        return ClassIterator(self)

# 对象必须提供一个next方法，执行该方法要么返回迭代器中的下一项，要么就引起一个stoplteration异常，以终止协议（只能往后走不能往前）
class ClassIterator(object):
    def __init__(self, obj):
        self.obj = obj
        self.current_num = 0

    def __iter__(self):
        pass
#  既有iter方法又有next方法则称之为“迭代器”
    def __next__(self):
        if self.current_num < len(self.obj.names):
            ret = self.obj.names[self.current_num]
            self.current_num += 1
            return ret
        else:
            raise StopIteration  # 抛出异常，自动停止循环

classmate = Classmate()
classmate.add('老王')
classmate.add('张三')
classmate.add('李四')

# print('判断classmate是否是可以迭代的对象：', isinstance(classmate, Iterable))  # 连续打印，直接逗号隔开
# classmate_iterator = iter(classmate)
# print('判断classmate_iterator是否是迭代器：', isinstance(classmate_iterator, Iterator))
# print(next(classmate_iterator))

# for循环机制的本质：循环所有对象，全部都是使用迭代器协议
for name in classmate:
    print(name)
    time.sleep(1)

迭代器二（迭代器与初始化合并一个方法）

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

class Classmate(object):
    def __init__(self):
        self.names = list()
        self.current_num = 0

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

    '''如果想要一个对象称为一个可以迭代的对象，即可以使用for，那么必须实现__iter__方法'''
    def __iter__(self):
        return self

    '''既有iter方法又有next方法则称之为迭代器'''
    def __next__(self):
        if self.current_num < len(self.names):
            ret = self.names[self.current_num]
            self.current_num += 1
            return ret
        else:
            raise StopIteration  # 抛出异常，自动停止循环

classmate = Classmate()
classmate.add('老王')
classmate.add('张三')
classmate.add('李四')

# print('判断classmate是否是可以迭代的对象：', isinstance(classmate, Iterable))  # 连续打印，直接逗号隔开
# classmate_iterator = iter(classmate)
# print('判断classmate_iterator是否是迭代器：', isinstance(classmate_iterator, Iterator))
# print(next(classmate_iterator))

# for循环机制的本质：循环所有对象，全部都是使用迭代器协议
for name in classmate:
    print(name)
    time.sleep(1)