fix_json.py

#!/usr/bin/env python2
#
# Copyright 2017 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
#
# pylint: disable=cros-logging-import

"""Script to re-format json result to one with branch_name and build_id"""
from __future__ import print_function

import argparse
import config
import json
import logging
import os
import subprocess
import sys

# Turn the logging level to INFO before importing other autotest
# code, to avoid having failed import logging messages confuse the
# test_droid user.
logging.basicConfig(level=logging.INFO)


def _parse_arguments_internal(argv):
  parser = argparse.ArgumentParser(description='Convert result to JSON'
                                   'format')
  parser.add_argument(
      '-b', '--bench', help='Generate JSON format file for which benchmark.')
  return parser.parse_args(argv)

def fix_json(bench):
  # Set environment variable for crosperf
  os.environ['PYTHONPATH'] = os.path.dirname(config.toolchain_utils)

  logging.info('Generating Crosperf Report...')
  json_path = os.path.join(config.bench_suite_dir, bench + '_refined')
  crosperf_cmd = [
      os.path.join(config.toolchain_utils, 'generate_report.py'), '--json',
      '-i=' + os.path.join(config.bench_suite_dir, bench + '.json'),
      '-o=' + json_path, '-f'
  ]

  # Run crosperf generate_report.py
  logging.info('Command: %s', crosperf_cmd)
  subprocess.call(crosperf_cmd)

  json_path += '.json'
  with open(json_path) as fout:
    objs = json.load(fout)
  for obj in objs:
    obj['branch_name'] = 'aosp/master'
    obj['build_id'] = 0
  with open(json_path, 'w') as fout:
    json.dump(objs, fout)

  logging.info('JSON file fixed successfully!')

def main(argv):
  arguments = _parse_arguments_internal(argv)

  bench = arguments.bench

  fix_json(bench)

if __name__ == '__main__':
  main(sys.argv[1:])