fizzbuzz

fizzbuzz.py

#!/usr/bin/env python3


def fizzbuzz():
    """ generator of fizzbuzz """
    i = 1
    while True:
        s = ''
        if i % 3 == 0:
            s += 'Fizz '
        if i % 5 == 0:
            s += 'Buzz'
        if s == '':
            s = str(i)
        s = s.rstrip()
        yield s
        i += 1


if __name__ == '__main__':
    from sys import argv
    num = argv[1]
    num = int(num)

    gen = fizzbuzz()
    for i in range(num):
        print(next(gen), end=', ')
    print()