1a0e92a802815d55dafd303b39843ddd2b9784bdbÉric Araujo#!/usr/bin/env python3 2e064b41f5ac046fc361fa80af551f5bfab01141cMartin v. Löwis""" Command line interface to difflib.py providing diffs in four formats: 3a33d17758974e0d150cc07a22c0c60b1485d0ab4Raymond Hettinger 4a33d17758974e0d150cc07a22c0c60b1485d0ab4Raymond Hettinger* ndiff: lists every line and highlights interline changes. 5e064b41f5ac046fc361fa80af551f5bfab01141cMartin v. Löwis* context: highlights clusters of changes in a before/after format. 6a33d17758974e0d150cc07a22c0c60b1485d0ab4Raymond Hettinger* unified: highlights clusters of changes in an inline format. 7e064b41f5ac046fc361fa80af551f5bfab01141cMartin v. Löwis* html: generates side by side comparison with change highlights. 8a33d17758974e0d150cc07a22c0c60b1485d0ab4Raymond Hettinger 9a33d17758974e0d150cc07a22c0c60b1485d0ab4Raymond Hettinger""" 10a33d17758974e0d150cc07a22c0c60b1485d0ab4Raymond Hettinger 11ccd047ea4b92f09a84b67e69deb82ce42e510c4cSerhiy Storchakaimport sys, os, difflib, argparse 12a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolskyfrom datetime import datetime, timezone 13a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky 14a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolskydef file_mtime(path): 15a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky t = datetime.fromtimestamp(os.stat(path).st_mtime, 16a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky timezone.utc) 17a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky return t.astimezone().isoformat() 18a33d17758974e0d150cc07a22c0c60b1485d0ab4Raymond Hettinger 19e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchlingdef main(): 20e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling 21992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser = argparse.ArgumentParser() 22992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser.add_argument('-c', action='store_true', default=False, 23992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka help='Produce a context format diff (default)') 24992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser.add_argument('-u', action='store_true', default=False, 25992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka help='Produce a unified format diff') 26992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser.add_argument('-m', action='store_true', default=False, 27992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka help='Produce HTML side by side diff ' 28992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka '(can use -c and -l in conjunction)') 29992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser.add_argument('-n', action='store_true', default=False, 30992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka help='Produce a ndiff format diff') 31992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser.add_argument('-l', '--lines', type=int, default=3, 32992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka help='Set number of context lines (default 3)') 33992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser.add_argument('fromfile') 34992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka parser.add_argument('tofile') 35992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka options = parser.parse_args() 36e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling 37e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling n = options.lines 38992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka fromfile = options.fromfile 39992334127e72c058cf5bd83ed458203e5ebfd74aSerhiy Storchaka tofile = options.tofile 40e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling 41a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky fromdate = file_mtime(fromfile) 42a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky todate = file_mtime(tofile) 436787a3806ee6a85a6f21ede70c10e15a6df267c4Serhiy Storchaka with open(fromfile) as ff: 44a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky fromlines = ff.readlines() 456787a3806ee6a85a6f21ede70c10e15a6df267c4Serhiy Storchaka with open(tofile) as tf: 46a2637729f23dc993e820fd92f0d1759ad714c9b2Alexander Belopolsky tolines = tf.readlines() 47e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling 48e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling if options.u: 49e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile, fromdate, todate, n=n) 50e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling elif options.n: 51e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling diff = difflib.ndiff(fromlines, tolines) 52e064b41f5ac046fc361fa80af551f5bfab01141cMartin v. Löwis elif options.m: 53e064b41f5ac046fc361fa80af551f5bfab01141cMartin v. Löwis diff = difflib.HtmlDiff().make_file(fromlines,tolines,fromfile,tofile,context=options.c,numlines=n) 54e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling else: 55e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling diff = difflib.context_diff(fromlines, tolines, fromfile, tofile, fromdate, todate, n=n) 56e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling 57e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling sys.stdout.writelines(diff) 58e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling 59e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchlingif __name__ == '__main__': 60e236b38731823aeb87bb6d101062f54a72044954Andrew M. Kuchling main() 61