Home | History | Annotate | Download | only in cli 
      1 #!/usr/bin/python
      2 
      3 import sys, optparse
      4 import common
      5 from autotest_lib.cli import rpc
      6 
      7 
      8 def parse_options():
      9     usage = "usage: %prog [options] job_id"
     10     parser = optparse.OptionParser(usage=usage)
     11     parser.add_option("-m", "--machine", dest="machine")
     12     parser.add_option("-t", "--test", dest="test")
     13     parser.add_option("-T", "--type", dest="type", type="choice",
     14                       choices=["all", "test", "iteration", "attr", "perf"])
     15     parser.add_option("-k", "--key", dest="key")
     16     parser.set_defaults(type="all")
     17     options, args = parser.parse_args()
     18     options.show_test_keyvals = options.type in ("all", "test")
     19     options.show_attr_keyvals = options.type in ("all", "iteration", "attr")
     20     options.show_perf_keyvals = options.type in ("all", "iteration", "perf")
     21     options.show_iter_keyvals = (
     22         options.show_perf_keyvals or options.show_attr_keyvals)
     23     return parser, options, args
     24 
     25 
     26 def print_keyvals(keyval, format_string, options):
     27     for key, value in keyval.iteritems():
     28         if not options.key or key == options.key:
     29             print format_string % (key, value)
     30 
     31 
     32 def print_views(test_views, options):
     33     for view in test_views:
     34         if not options.machine:
     35             print "Machine: %s" % view["hostname"]
     36         if not options.test:
     37             print "Test: %s" % view["test_name"]
     38         if options.show_test_keyvals:
     39             print "Test Attributes:"
     40             print_keyvals(view["attributes"], "\t%s = %s", options)
     41         if options.show_iter_keyvals:
     42             print "Iteration Attributes:"
     43             for i, iteration in enumerate(view["iterations"]):
     44                 print "\tIteration #%d:" % (i + 1)
     45                 if options.show_attr_keyvals:
     46                     print_keyvals(iteration["attr"], "\t\t%s(attr) = %s",
     47                                   options)
     48                 if options.show_perf_keyvals:
     49                     print_keyvals(iteration["perf"], "\t\t%s(perf) = %s",
     50                                   options)
     51         print
     52 
     53 
     54 def main():
     55     parser, options, args = parse_options()
     56     if not args:
     57         parser.print_help()
     58         return
     59 
     60     query_filter = {}
     61     if options.machine:
     62         query_filter["hostname"] = options.machine
     63     if options.test:
     64         query_filter["test_name"] = options.test
     65 
     66     comm = rpc.tko_comm()
     67     test_views = []
     68     for job_id in args:
     69         query_filter["job_tag__startswith"] = "%s-" % job_id
     70         test_views += comm.run("get_detailed_test_views", **query_filter)
     71     print_views(test_views, options)
     72 
     73 
     74 if __name__ == "__main__":
     75     main()
     76