comparison hglib/util.py @ 9:5882a698ad5c

util: rewrite eatlines (faster and simpler version)

8:3ac38d500d68

import itertools

def grouper(n, iterable):
    ''' list(grouper(2, range(4))) -> [(0, 1), (2, 3)] '''
    args = [iter(iterable)] * n
    return itertools.izip(*args)

def eatlines(s, n):
    idx = 0
    for i in xrange(n):
        idx = s.find('\n', idx) + 1

    return s[idx:]

def cmdbuilder(name, *args, **kwargs):
    """
    A helper for building the command arguments

9:5882a698ad5c

import itertools, cStringIO

def grouper(n, iterable):
    ''' list(grouper(2, range(4))) -> [(0, 1), (2, 3)] '''
    args = [iter(iterable)] * n
    return itertools.izip(*args)

def eatlines(s, n):
    """
    >>> eatlines("1\\n2", 1)
    '2'
    >>> eatlines("1\\n2", 2)
    ''
    >>> eatlines("1\\n2", 3)
    ''
    >>> eatlines("1\\n2\\n3", 1)
    '2\\n3'
    """
    cs = cStringIO.StringIO(s)

    for line in cs:
        n -= 1
        if n == 0:
            return cs.read()
    return ''

def cmdbuilder(name, *args, **kwargs):
    """
    A helper for building the command arguments