diff --git a/utils/diff_db_bits.py b/utils/diff_db_bits.py new file mode 100755 index 00000000..083f9258 --- /dev/null +++ b/utils/diff_db_bits.py @@ -0,0 +1,44 @@ +#!/usr/bin/env python3 +import argparse +import subprocess +import tempfile +import os.path +import glob + + +def main(): + parser = argparse.ArgumentParser( + description= + "Tool for comparing database segbits outputs from two database's.") + parser.add_argument('a_db') + parser.add_argument('b_db') + + args = parser.parse_args() + + assert os.path.isdir(args.a_db) + assert os.path.isdir(args.b_db) + + for a_db_in in glob.glob('{}/segbits*.db'.format(args.a_db)): + a_db_base = os.path.basename(a_db_in) + b_db_in = '{}/{}'.format(args.b_db, a_db_base) + + if not os.path.exists(b_db_in): + print('{} not found!'.format(b_db_in)) + continue + + with tempfile.NamedTemporaryFile(suffix="_a_{}".format( + a_db_base)) as a_db_out, tempfile.NamedTemporaryFile( + suffix="_b_{}".format(a_db_base)) as b_db_out: + subprocess.check_call( + "sort {}".format(a_db_in), shell=True, stdout=a_db_out) + subprocess.check_call( + "sort {}".format(b_db_in), shell=True, stdout=b_db_out) + + print("Comparing {}".format(a_db_base)) + subprocess.call( + "diff -U 0 {} {}".format(a_db_out.name, b_db_out.name), + shell=True) + + +if __name__ == "__main__": + main()