for循环的底层原理

# 自定义可迭代对象：在类里面提供了__iter__方法创建的对象就是可迭代对象
# 自定义迭代器对象：在类里面提供__iter__和__next__的方法创建的对象就是迭代器对象
# 迭代器的作用： 记录当前数据的位置以后获取下一个位置的值
# iter函数:获取可迭代对象的迭代器， 会调用可迭代对象身上的__iter__方法
# next函数:获取迭代器中下一个值，会调用迭代器对象身上的__next__方法
# for 循环的本质: 如果遍历的对象是可迭代对象那么会通过iter函数获取迭代器，然后通过next函数获取迭代器中下一个值
# 如果遍历的对象是迭代器，那么会通过next函数获取迭代器中下一个值
# 提醒: for循环内部自己捕获停止迭代的异常， 使用while True 取值自己需要捕获停止迭代异常


# 自定义可迭代对象
class Student_List(object):

    # 提供构造方法，创建存储数据的容器->列表
    def __init__(self):
        self.my_list = []

    # 向列表中添加学生信息
    def add_student(self,student):
        self.my_list.append(student)

    # 提供可以迭代的方法
    def __iter__(self):
        student_iterator = Student_Iterator(self.my_list)
        return student_iterator



# 自定义迭代器
class Student_Iterator(object):
    # 把自定义的可迭代对象传过来
    def __init__(self, my_list):
        self.my_list = my_list
        # 设置默认下标为0
        self.current_index = 0

    # 提供可以迭代的方法,返回自身即可
    def __iter__(self):
        return self

    # 采用__next__()方法获取对象中下一个值
    def __next__(self):
        # 如果下标小于列表长度，则返回取到相应下标列表的值
        if self.current_index < len(self.my_list):
            # 要在return之前将下标值加1，因为若放在return之后就不执行了
            self.current_index += 1
            # 因为先将下标进行了加1操作，所有这里要减1
            return self.my_list[self.current_index-1]
        else:
            print('下标越界')
            raise StopIteration

if __name__ == '__main__':
    student_list = Student_List()
    student_list.add_student('ss')
    student_list.add_student('xx')

# 遍历
    for student in student_list:
        print(student)