General Fizzbuzz III

fizzbuzz.py

def n_of_k(n, k):

    return lambda n: n % k == 0


def n_for_ks(n, *ks):

    return [n_of_k(n, k) for k in ks]


def fizzbuzz(rules, data):

    def make(s):
        return lambda n: s(n) if callable(s) else s

    rules = [(r, make(s)) for r, s in rules]

    for n in data:
        for i, c in enumerate(rules):

            conds, fn = c

            if all([f(n) for f in n_for_ks(n, *conds)]):
                yield (fn(n))
                break


def test():

    rules = [
        ([3, 5, 2], "WOW!"),
        ([3, 5], "FizzBuzz"),
        ([3], "Fizz"),
        ([5], "Buzz"),
        ([1], lambda n: n)
    ]

    data = xrange(1, 31)

    fizzbuzz_1 = fizzbuzz(rules, data)

    for result in fizzbuzz_1:
        print (result)

test()