Testing (creating) custom python iterators using iter, next, or getitem

custom_python_iters.py

# Testing __getitem__() and Python Iterators
"""
    Some resources:
    - https://stackoverflow.com/a/61658447/5836037
"""

# %%
class Something:
    def __init__(self) -> None:
        self.items = [10, 20, 35, 60, 75]
        self.ki = 0
    
    def __getitem__(self, i):
        return self.items[i]

    def __iter__(self):
        print(f"Iterator called, resetting 'ki'")
        self.ki = 0
        return self
    
    def __next__(self):
        if self.ki >= len(self.items):
            raise StopIteration("Stop the iter")
        ret = self.items[self.ki]
        self.ki += 1
        return ret

# %%
a = Something()

# %%
i = 0
for k in a:
    print(k)
    i += 1
    if i > 20:
        break

# %%
ai = iter(a)
while True:
    print(next(ai))
# The last result should be a StopIteration exception

# %%