diff-trunc.py

#! /usr/bin/env python
"""
Display a unified diff between two files, for use w/cmdrun and 'diff' hilite.
"""
import sys
import argparse
from difflib import unified_diff


def main():
    p = argparse.ArgumentParser()
    p.add_argument('file1')
    p.add_argument('file2')
    p.add_argument('-o', '--output')
    args = p.parse_args()

    data1 = open(args.file1, 'rt').readlines()
    data2 = open(args.file2, 'rt').readlines()

    outfp = sys.stdout
    if args.output:
        outfp = open(args.output, 'wt')

    x = list(unified_diff(data1, data2))
    for i in x[3:]:
        outfp.write(i)


if __name__ == '__main__':
    sys.exit(main())