Convert shell script to python script.

Signed-off-by: Keith Rothman <[email protected]>
This commit is contained in:
Keith Rothman
2019-01-23 13:09:19 -08:00
parent 5817e5932c
commit 782d2c4ced
2 changed files with 44 additions and 26 deletions
+44
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()