Merge pull request #553 from litghost/diff_db_bits

Add utility for diffing segbits from two databases.

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()