Custom iterator object examples.

iterable_custom_loop_limit.py

class MyIterator:

    def __init__(self, limit):
        self.limit = limit

    def __iter__(self):
        self.a = 1
        return self

    def __next__(self):
        if self.a <= self.limit:
            x = self.a
            self.a += 1
            return x
        else:
            raise StopIteration


obj = MyIterator(10)
my_iter = iter(obj)

for x in my_iter:
    print(f"Step: {x}")

iteration_iter_custom.py

class MyIterator:

    def __iter__(self):
        self.a = 1
        return self

    def __next__(self):
        x = self.a
        self.a += 1
        return x


obj = MyIterator()
my_iter = iter(obj)

print(next(my_iter))
print(next(my_iter))
print(next(my_iter))
print(next(my_iter))
print(next(my_iter))

iteration_iter_custom_loop.py

class MyIterator:

    def __iter__(self):
        self.a = 1
        return self

    def __next__(self):
        x = self.a
        self.a += 1
        return x


obj = MyIterator()
my_iter = iter(obj)

for i in my_iter:
    print(i)
    if i == 10:
        print("This iteration will go forever if we don't stop it...")
        break