iterator_with_return.py

def iterator(value):
    if value:
        return 10
    else:
        for i in range(5):
            yield i

print(iterator(True), iterator(False))

for x in iterator(True):
    print(x)

for x in iterator(False):
    print(x)

# Output
# <generator object iterator at 0x7f0571587eb8> <generator object iterator at 0x7f0571587f10>
# 0
# 1
# 2
# 3
# 4