Counters, Closures and Generators

generator.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import collections

__author__ = 'Andrew W. Donoho, <github@store.ddg.com>'
__copyright__ = 'Public Domain'


def counter() -> collections.Iterator:

    def count() -> int:

        n = 0

        while n < 10:

            n += 1

            yield n

    return iter(count())


def counter2(count: int) -> collections.Iterator:

    n = 0

    while n < count:

        n += 1

        yield n


def main():

    for i in counter():

        print(i)

    for i in counter2(5):

        print(i)


if __name__ == '__main__':

    main()