split_on_packets

gistfile1.py

def split_on_packets(iterable, packet_size):
    '''
    >>> a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
    >>> list(split_on_packets(a, 3))
    [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11]]

    >>> list(split_on_packets(a, 13))
    [[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]]

    >>> list(split_on_packets(a, 1))
    [[1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11]]
    '''
    assert(hasattr(iterable, '__iter__'))
    assert(packet_size > 0)
    
    iter = iterable.__iter__()
    while True:
        packet = []
        try:
            for i in xrange(packet_size):
                packet.append(iter.next())
            yield packet
        except StopIteration:
            if packet:
                yield packet
            break