• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python3
2""" Command line interface to difflib.py providing diffs in four formats:
3
4* ndiff:    lists every line and highlights interline changes.
5* context:  highlights clusters of changes in a before/after format.
6* unified:  highlights clusters of changes in an inline format.
7* html:     generates side by side comparison with change highlights.
8
9"""
10
11import sys, os, difflib, argparse
12from datetime import datetime, timezone
13
14def file_mtime(path):
15    t = datetime.fromtimestamp(os.stat(path).st_mtime,
16                               timezone.utc)
17    return t.astimezone().isoformat()
18
19def main():
20
21    parser = argparse.ArgumentParser()
22    parser.add_argument('-c', action='store_true', default=False,
23                        help='Produce a context format diff (default)')
24    parser.add_argument('-u', action='store_true', default=False,
25                        help='Produce a unified format diff')
26    parser.add_argument('-m', action='store_true', default=False,
27                        help='Produce HTML side by side diff '
28                             '(can use -c and -l in conjunction)')
29    parser.add_argument('-n', action='store_true', default=False,
30                        help='Produce a ndiff format diff')
31    parser.add_argument('-l', '--lines', type=int, default=3,
32                        help='Set number of context lines (default 3)')
33    parser.add_argument('fromfile')
34    parser.add_argument('tofile')
35    options = parser.parse_args()
36
37    n = options.lines
38    fromfile = options.fromfile
39    tofile = options.tofile
40
41    fromdate = file_mtime(fromfile)
42    todate = file_mtime(tofile)
43    with open(fromfile) as ff:
44        fromlines = ff.readlines()
45    with open(tofile) as tf:
46        tolines = tf.readlines()
47
48    if options.u:
49        diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile, fromdate, todate, n=n)
50    elif options.n:
51        diff = difflib.ndiff(fromlines, tolines)
52    elif options.m:
53        diff = difflib.HtmlDiff().make_file(fromlines,tolines,fromfile,tofile,context=options.c,numlines=n)
54    else:
55        diff = difflib.context_diff(fromlines, tolines, fromfile, tofile, fromdate, todate, n=n)
56
57    sys.stdout.writelines(diff)
58
59if __name__ == '__main__':
60    main()
61