1# resulttool - merge multiple testresults.json files into a file or directory
2#
3# Copyright (c) 2019, Intel Corporation.
4# Copyright (c) 2019, Linux Foundation
5#
6# SPDX-License-Identifier: GPL-2.0-only
7#
8
9import os
10import json
11import resulttool.resultutils as resultutils
12
13def merge(args, logger):
14    configvars = {}
15    if not args.not_add_testseries:
16        configvars = resultutils.extra_configvars.copy()
17    if args.executed_by:
18        configvars['EXECUTED_BY'] = args.executed_by
19    if resultutils.is_url(args.target_results) or os.path.isdir(args.target_results):
20        results = resultutils.load_resultsdata(args.target_results, configmap=resultutils.store_map, configvars=configvars)
21        resultutils.append_resultsdata(results, args.base_results, configmap=resultutils.store_map, configvars=configvars)
22        resultutils.save_resultsdata(results, args.target_results)
23    else:
24        results = resultutils.load_resultsdata(args.base_results, configmap=resultutils.flatten_map, configvars=configvars)
25        if os.path.exists(args.target_results):
26            resultutils.append_resultsdata(results, args.target_results, configmap=resultutils.flatten_map, configvars=configvars)
27        resultutils.save_resultsdata(results, os.path.dirname(args.target_results), fn=os.path.basename(args.target_results))
28
29    logger.info('Merged results to %s' % os.path.dirname(args.target_results))
30
31    return 0
32
33def register_commands(subparsers):
34    """Register subcommands from this plugin"""
35    parser_build = subparsers.add_parser('merge', help='merge test result files/directories/URLs',
36                                         description='merge the results from multiple files/directories/URLs into the target file or directory',
37                                         group='setup')
38    parser_build.set_defaults(func=merge)
39    parser_build.add_argument('base_results',
40                              help='the results file/directory/URL to import')
41    parser_build.add_argument('target_results',
42                              help='the target file or directory to merge the base_results with')
43    parser_build.add_argument('-t', '--not-add-testseries', action='store_true',
44                              help='do not add testseries configuration to results')
45    parser_build.add_argument('-x', '--executed-by', default='',
46                              help='add executed-by configuration to each result file')
47