flatten.py

import collections
import itertools

def flattened(i):
    i = iter(i)
    while True:
        first = i.next()
        if isinstance(first, collections.Iterable):
            i = itertools.chain(first, i)
        else:
            yield i