Merge pull request #552 from litghost/add_check_json_utility

Add utility to compare JSON from two database's.
This commit is contained in:
litghost 2019-01-23 17:37:58 -08:00 committed by GitHub
commit c595847739
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 44 additions and 0 deletions

44
utils/diff_db_json.py Executable file
View File

@ -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 JSON 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_json_in in glob.glob('{}/*.json'.format(args.a_db)):
a_json_base = os.path.basename(a_json_in)
b_json_in = '{}/{}'.format(args.b_db, a_json_base)
if not os.path.exists(b_json_in):
print('{} not found!'.format(b_json_in))
continue
with tempfile.NamedTemporaryFile(suffix="_a_{}".format(
a_json_base)) as a_json_out, tempfile.NamedTemporaryFile(
suffix="_b_{}".format(a_json_base)) as b_json_out:
subprocess.check_call(
"python3 -m utils.xjson {}".format(a_json_in), shell=True, stdout=a_json_out)
subprocess.check_call(
"python3 -m utils.xjson {}".format(b_json_in), shell=True, stdout=b_json_out)
print("Comparing {}".format(a_json_base))
subprocess.call(
"diff -U 3 {} {}".format(a_json_out.name, b_json_out.name),
shell=True)
if __name__ == "__main__":
main()