python_iteration_4.py

python_iteration_4.py

class MyIterator:
    def __init__(self, max_num):
        self.max_num = max_num

    def __getitem__(self, key):
        if key <= self.max_num:
            return key
        else:
            raise IndexError

my_iterator = MyIterator(3)

for item in my_iterator:
    print(item)
# 0
# 1
# 2
# 3