| #!/usr/bin/env python3 | |
| # -*- coding: utf-8 -*- | |
| # | |
| # Copyright (C) 2017-2020 The Project X-Ray Authors. | |
| # | |
| # Use of this source code is governed by a ISC-style | |
| # license that can be found in the LICENSE file or at | |
| # https://opensource.org/licenses/ISC | |
| # | |
| # SPDX-License-Identifier: ISC | |
| 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() |