comparison mercurial/patch.py @ 37471:51d5e1ff0613

py3: use s.startswith() instead of s[n] while parsing patches

I know 'bytes[n] in bytes' magically works, but I'm tired of finding which
one breaks the tests.

37470:d658cbef8041

# public functions

def split(stream):
    '''return an iterator of individual patches from a stream'''
    def isheader(line, inheader):
        if inheader and line[0] in (' ', '\t'):
            # continuation
            return True
        if line[0] in (' ', '-', '+'):
            # diff line - don't check for header pattern in there
            return False
        l = line.split(': ', 1)
        return len(l) == 2 and ' ' not in l[0]

            top = 0
            bot = 0
            hlen = len(self.hunk)
            for x in xrange(hlen - 1):
                # the hunk starts with the @@ line, so use x+1
                if self.hunk[x + 1][0] == ' ':
                    top += 1
                else:
                    break
            if not toponly:
                for x in xrange(hlen - 1):
                    if self.hunk[hlen - bot - 1][0] == ' ':
                        bot += 1
                    else:
                        break

            bot = min(fuzz, bot)

                tofile = lr.readline()
                header += [fromfile, tofile]
            else:
                lr.push(fromfile)
            yield 'file', header
        elif line[0:1] == ' ':
            yield 'context', scanwhile(line, lambda l: l[0] in ' \\')
        elif line[0] in '-+':
            yield 'hunk', scanwhile(line, lambda l: l[0] in '-+\\')
        else:
            m = lines_re.match(line)
            if m:
                yield 'range', m.groups()
            else:

                yield ('\n', '')
            if head:
                if line.startswith('@'):
                    head = False
            else:
                if line and line[0] not in ' +-@\\':
                    head = True
            stripline = line
            diffline = False
            if not head and line and line[0] in '+-':
                # highlight tabs and trailing whitespace, but only in
                # changed lines
                stripline = line.rstrip()
                diffline = True

    lastmatch = 0
    matches = {}
    for i, line in enumerate(slist):
        if line == '':
            continue
        if line[0] == '-':
            lastmatch = max(lastmatch, i)
            newgroup = False
            for j, newline in enumerate(slist[lastmatch + 1:]):
                if newline == '':
                    continue
                if newline[0] == '-' and newgroup: # too far, no match
                    break
                if newline[0] == '+': # potential match
                    newgroup = True
                    sim = difflib.SequenceMatcher(None, line, newline).ratio()
                    if sim > 0.7:
                        lastmatch = lastmatch + 1 + j
                        matches[i] = lastmatch

                        break
    return matches

def _inlinediff(s1, s2, operation):
    '''Perform string diff to highlight specific changes.'''
    operation_skip = '+?' if operation == 'diff.deleted' else '-?'
    if operation == 'diff.deleted':
        s2, s1 = s1, s2

    buff = []
    # we never want to higlight the leading +-

        raise error.ProgrammingError("Case not expected, operation = %s" %
                                     operation)

    s = difflib.ndiff(_nonwordre.split(s2), _nonwordre.split(s1))
    for part in s:
        if part[0] in operation_skip or len(part) == 2:
            continue
        l = operation + '.highlight'
        if part[0] in ' ':
            l = operation
        if part[2:] == '\t':
            l = 'diff.tab'
        if l == label: # contiguous token with same label
            token += part[2:]

37471:51d5e1ff0613

# public functions

def split(stream):
    '''return an iterator of individual patches from a stream'''
    def isheader(line, inheader):
        if inheader and line.startswith((' ', '\t')):
            # continuation
            return True
        if line.startswith((' ', '-', '+')):
            # diff line - don't check for header pattern in there
            return False
        l = line.split(': ', 1)
        return len(l) == 2 and ' ' not in l[0]

            top = 0
            bot = 0
            hlen = len(self.hunk)
            for x in xrange(hlen - 1):
                # the hunk starts with the @@ line, so use x+1
                if self.hunk[x + 1].startswith(' '):
                    top += 1
                else:
                    break
            if not toponly:
                for x in xrange(hlen - 1):
                    if self.hunk[hlen - bot - 1].startswith(' '):
                        bot += 1
                    else:
                        break

            bot = min(fuzz, bot)

                tofile = lr.readline()
                header += [fromfile, tofile]
            else:
                lr.push(fromfile)
            yield 'file', header
        elif line.startswith(' '):
            cs = (' ', '\\')
            yield 'context', scanwhile(line, lambda l: l.startswith(cs))
        elif line.startswith(('-', '+')):
            cs = ('-', '+', '\\')
            yield 'hunk', scanwhile(line, lambda l: l.startswith(cs))
        else:
            m = lines_re.match(line)
            if m:
                yield 'range', m.groups()
            else:

                yield ('\n', '')
            if head:
                if line.startswith('@'):
                    head = False
            else:
                if line and not line.startswith((' ', '+', '-', '@', '\\')):
                    head = True
            stripline = line
            diffline = False
            if not head and line and line.startswith(('+', '-')):
                # highlight tabs and trailing whitespace, but only in
                # changed lines
                stripline = line.rstrip()
                diffline = True

    lastmatch = 0
    matches = {}
    for i, line in enumerate(slist):
        if line == '':
            continue
        if line.startswith('-'):
            lastmatch = max(lastmatch, i)
            newgroup = False
            for j, newline in enumerate(slist[lastmatch + 1:]):
                if newline == '':
                    continue
                if newline.startswith('-') and newgroup: # too far, no match
                    break
                if newline.startswith('+'): # potential match
                    newgroup = True
                    sim = difflib.SequenceMatcher(None, line, newline).ratio()
                    if sim > 0.7:
                        lastmatch = lastmatch + 1 + j
                        matches[i] = lastmatch

                        break
    return matches

def _inlinediff(s1, s2, operation):
    '''Perform string diff to highlight specific changes.'''
    operation_skip = ('+', '?') if operation == 'diff.deleted' else ('-', '?')
    if operation == 'diff.deleted':
        s2, s1 = s1, s2

    buff = []
    # we never want to higlight the leading +-

        raise error.ProgrammingError("Case not expected, operation = %s" %
                                     operation)

    s = difflib.ndiff(_nonwordre.split(s2), _nonwordre.split(s1))
    for part in s:
        if part.startswith(operation_skip) or len(part) == 2:
            continue
        l = operation + '.highlight'
        if part.startswith(' '):
            l = operation
        if part[2:] == '\t':
            l = 'diff.tab'
        if l == label: # contiguous token with same label
            token += part[2:]