Flatten arbitrary nested lists

flatten.py

def flatten(iterable):
    """Return a generator that flattens an iterable
    
    Safe to use on nested iterables.
    
    Args:
        iterable: any iterable.
        
    Returns:
        generator of all items in iterable
        
    Example:
        >>> list(flatten([1, [2, 3], [4, [5, 6]]]))
        [1, 2, 3, 4, 5, 6]
    """
    for element in iterable:
        try:
            yield from flatten(element)
        except TypeError:
            yield element