pinpoint_cli

#!/usr/bin/env vpython
# Copyright 2018 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import json
import os
import sys

from core import path_util
path_util.AddPyUtilsToPath()
path_util.AddTracingToPath()

from core import cli_utils
from core import external_modules
from cli_tools.pinpoint_cli import commands
from core.services import luci_auth


def main():
  parser = argparse.ArgumentParser()
  parser.add_argument(
      '-v', '--verbose', action='count', default=0,
      help='Increase verbosity level')
  subparsers = parser.add_subparsers(dest='action')
  subparsers.required = True

  subparser = subparsers.add_parser(
      'status', help='check the status of some pinpoint jobs')
  subparser.add_argument(
      'job_ids', metavar='JOB_ID', nargs='+',
      help='one or more pinpoint job ids')

  subparser = subparsers.add_parser(
      'get-csv', help='download the perf results from jobs as a csv file')
  subparser.add_argument(
      '--only-differences', action='store_true',
      help='on bisect jobs, only get data for changes immediately before/after'
           ' differences in the comparison metric.')
  subparser.add_argument(
      '--output', metavar='OUTPUT_CSV', default='job_results.csv',
      help='path to a file where to store perf results as a csv file'
           ' (default: %(default)s)')
  subparser.add_argument(
      'job_ids', metavar='JOB_ID', nargs='+',
      help='one or more pinpoint job ids')

  subparser = subparsers.add_parser(
      'start-job', help='start a new pinpoint job')
  subparser.add_argument(
      'config_path', metavar='CONFIG_PATH',
      help='path to a json file with a pinpoint job configuration (see'
           " examples/pinpoint_cli directory) or '-' to read it from stdin")
  args = parser.parse_args()
  cli_utils.ConfigureLogging(args.verbose)

  luci_auth.CheckLoggedIn()
  if args.action == 'status':
    return commands.CheckJobStatus(args.job_ids)
  elif args.action == 'get-csv':
    return commands.DownloadJobResultsAsCsv(
        args.job_ids, args.only_differences, args.output)
  elif args.action == 'start-job':
    return commands.StartJobFromConfig(args.config_path)
  else:
    raise NotImplementedError(args.action)


if __name__ == '__main__':
  external_modules.RequireModules()
  sys.exit(main())