tools/crunchstat-summary/crunchstat_summary/command.py

   1 import argparse
   2 import gzip
   3 import sys
   4
   5 from crunchstat_summary import summarizer
   6
   7
   8 class ArgumentParser(argparse.ArgumentParser):
   9     def __init__(self):
  10         super(ArgumentParser, self).__init__(
  11             description='Summarize resource usage of an Arvados Crunch job')
  12         src = self.add_mutually_exclusive_group()
  13         src.add_argument(
  14             '--job', type=str, metavar='UUID',
  15             help='Look up the specified job and read its log data from Keep')
  16         src.add_argument(
  17             '--pipeline-instance', type=str, metavar='UUID',
  18             help='Summarize each component of the given pipeline instance')
  19         src.add_argument(
  20             '--log-file', type=str,
  21             help='Read log data from a regular file')
  22
  23
  24 class Command(object):
  25     def __init__(self, args):
  26         self.args = args
  27
  28     def summarizer(self):
  29         if self.args.pipeline_instance:
  30             return summarizer.PipelineSummarizer(self.args.pipeline_instance)
  31         elif self.args.job:
  32             return summarizer.JobSummarizer(self.args.job)
  33         elif self.args.log_file:
  34             if self.args.log_file.endswith('.gz'):
  35                 fh = gzip.open(self.args.log_file)
  36             else:
  37                 fh = open(self.args.log_file)
  38             return summarizer.Summarizer(fh)
  39         else:
  40             return summarizer.Summarizer(sys.stdin)