flatten.py

def flatten (*args):
    for x in args:
        if hasattr(x, '__iter__'):
            for y in flatten(*x):
                yield y
        else:
            yield x

# flatten(1, 2, [3, 4, 5]) = [1, 2, 3, 4, 5]