fizzbuzz

fizzbuzz.py

class FizzBuzzGenerator(object):
    """docstring for Generator."""

    fizz_mapper = {
        3: "fizz",
        5: "buzz",
        7: "fooo",
    }

    def __init__(self, count=5):
        self.count = count + 1

    def fizz_buzz(self, input):
        res = []

        for k, v in self.fizz_mapper.items():
            if input % k == 0:
                res.append(v)

        output = " ".join(res)
        return output or input

    def run(self):
        return (self.fizz_buzz(x) for x in xrange(1, self.count))

fbg = FizzBuzzGenerator(105)
for i in fbg.run():
    print(i)