diff options
Diffstat (limited to 'scripts/lib/resulttool/merge.py')
-rw-r--r-- | scripts/lib/resulttool/merge.py | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/scripts/lib/resulttool/merge.py b/scripts/lib/resulttool/merge.py new file mode 100644 index 0000000000..1d9cfafd41 --- /dev/null +++ b/scripts/lib/resulttool/merge.py @@ -0,0 +1,71 @@ +# test result tool - merge multiple testresults.json files +# +# Copyright (c) 2019, Intel Corporation. +# +# This program is free software; you can redistribute it and/or modify it +# under the terms and conditions of the GNU General Public License, +# version 2, as published by the Free Software Foundation. +# +# This program is distributed in the hope it will be useful, but WITHOUT +# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or +# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for +# more details. +# +from resulttool.resultsutils import load_json_file, get_dict_value, dump_json_data +import os +import json + +class ResultsMerge(object): + + def get_test_results(self, logger, file, result_id): + results = load_json_file(file) + if result_id: + result = get_dict_value(logger, results, result_id) + if result: + return {result_id: result} + return result + return results + + def merge_results(self, base_results, target_results): + for k in target_results: + base_results[k] = target_results[k] + return base_results + + def _get_write_dir(self): + basepath = os.environ['BUILDDIR'] + return basepath + '/tmp/' + + def dump_merged_results(self, results, output_dir): + file_output_dir = output_dir if output_dir else self._get_write_dir() + dump_json_data(file_output_dir, 'testresults.json', results) + print('Successfully merged results to: %s' % os.path.join(file_output_dir, 'testresults.json')) + + def run(self, logger, base_result_file, target_result_file, target_result_id, output_dir): + base_results = self.get_test_results(logger, base_result_file, '') + target_results = self.get_test_results(logger, target_result_file, target_result_id) + if base_results and target_results: + merged_results = self.merge_results(base_results, target_results) + self.dump_merged_results(merged_results, output_dir) + +def merge(args, logger): + merge = ResultsMerge() + merge.run(logger, args.base_result_file, args.target_result_file, args.target_result_id, args.output_dir) + return 0 + +def register_commands(subparsers): + """Register subcommands from this plugin""" + parser_build = subparsers.add_parser('merge', help='merge test results', + description='merge results from multiple files', + group='setup') + parser_build.set_defaults(func=merge) + parser_build.add_argument('base_result_file', + help='base result file provide the base result set') + parser_build.add_argument('target_result_file', + help='target result file provide the target result set for merging into the ' + 'base result set') + parser_build.add_argument('-t', '--target-result-id', default='', + help='(optional) default merge all result sets available from target to base ' + 'unless specific target result id was provided') + parser_build.add_argument('-o', '--output-dir', default='', + help='(optional) default write merged results to <poky>/build/tmp/ unless specific ' + 'output directory was provided') |