view en/examples/data/check_whitespace.py @ 49:18210d46491f

More hook content.

#!/usr/bin/python

import re

def trailing_whitespace(difflines):
    added, linenum, header = [], 0, False

    for line in difflines:
        if header:
            if line.startswith('+++ '):
                header = False
            else:
                # remember the name of the file that this diff affects
                m = re.match(r'--- [^/]/([^\t])', line)
                if m: filename = m.group(1)
            continue
        if line.startswith('diff '):
            header = True
            continue
        # hunk header - save the line number
        m = re.match(r'@@ -(\d+),', line)
        if m:
            linenum = int(m.group(1))
            continue
        # hunk body - check for an added line with trailing whitespace
        m = re.match(r'\+.*\s$', line)
        if m:
            added.append((filename, linenum))
        if line and line[0] in ' +':
            linenum += 1
    return added

if __name__ == '__main__':
    import os, sys
    
    added = trailing_whitespace(os.popen('hg export tip'))
    if added:
        for filename, linenum in added:
            print >> sys.stderr, ('%s, line %d: trailing whitespace added' %
                                  (filename, linenum))
        # save the commit message so we don't need to retype it
        os.system('hg tip --template "{desc}" > .hg/commit.save')
        print >> sys.stderr, 'commit message saved to .hg/commit.save'
        sys.exit(1)