version_compare.py 1.66 KB
Newer Older
Anders Blomdell's avatar
Anders Blomdell committed
1 2 3 4 5 6 7 8 9 10 11
#!/usr/bin/python

import os
import sys
import difflib
import re

class File:

    def __init__(self, path, match, replacement):
        def replace(s):
12
            return re.sub('[ \t]+', ' ', s).replace(match, replacement).strip()
Anders Blomdell's avatar
Anders Blomdell committed
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
        self.name = path.replace(match, replacement)
        self.path = path
        with open(path) as f:
            self.lines = map(replace, f.readlines())

    def __cmp__(self, other):
        if other == None:
            return cmp(self.name, other)
        return cmp(self.name, other.name)

def readfiles(root, match, replacement):
    result = []
    for n in os.listdir(root):
        path = os.path.join(root, n)
        if os.path.islink(path):
            pass
        elif os.path.isdir(path):
            for f in filter(None, readfiles(path, match, replacement)):
                result.append(f)
        else:
            result.append(File(path, match, replacement))
    for f in sorted(result):
        yield f
    yield None

if __name__ == '__main__':
    A = readfiles(*sys.argv[1:4])
    B = readfiles(*sys.argv[4:7])
    a = A.next()
    b = B.next()
    while a != None or b != None:
        if b == None or a.name < b.name:
            print "Only in %s:" %sys.argv[1], a.path
            a = A.next()
        elif a == None or a.name > b.name: 
            print "Only in %s:" %sys.argv[4], b.path
            b = B.next()
        else:
Anders Blomdell's avatar
Anders Blomdell committed
51 52
            equal = True
            for l in difflib.unified_diff(a.lines, b.lines, a.path, b.path):
53
                print l
Anders Blomdell's avatar
Anders Blomdell committed
54 55 56 57
                equal = False
            if equal:
                print "Identical", a.path, b.path
                
Anders Blomdell's avatar
Anders Blomdell committed
58 59
            a = A.next() 
            b = B.next()