comparison hglib/util.py @ 19:19d2c55c3928

util: introduce skiplines

18:518149e32888

    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

19:19d2c55c3928

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

def skiplines(s, prefix):
    """
    Skip lines starting with prefix in s

    >>> skiplines('a\\nb\\na\\n', 'a')
    'b\\na\\n'
    >>> skiplines('a\\na\\n', 'a')
    ''
    >>> skiplines('', 'a')
    ''
    >>> skiplines('a\\nb', 'b')
    'a\\nb'
    """
    cs = cStringIO.StringIO(s)

    for line in cs:
        if not line.startswith(prefix):
            return line + cs.read()

    return ''

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