prjxray/utils/diff_db_bits.py

45 lines
1.3 KiB
Python
Raw Normal View History

#!/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()