client/tests/kvm/scan_results.py

#!/usr/bin/env python
"""
Program that parses the autotest results and return a nicely printed final test
result.

@copyright: Red Hat 2008-2009
"""

def parse_results(text):
    """Parse text containing Autotest results and return a list of result 4-tuples."""
    result_list = []
    start_time_list = []
    info_list = []

    lines = text.splitlines()
    for line in lines:
        line = line.strip()
        parts = line.split("\t")

        # Found a START line -- get start time
        if line.startswith("START") and len(parts) >= 5 and parts[3].startswith("timestamp"):
            start_time = float(parts[3].split('=')[1])
            start_time_list.append(start_time)
            info_list.append("")

        # Found an END line -- get end time, name and status
        elif line.startswith("END") and len(parts) >= 5 and parts[3].startswith("timestamp"):
            end_time = float(parts[3].split('=')[1])
            start_time = start_time_list.pop()
            info = info_list.pop()
            test_name = parts[2]
            test_status = parts[0].split()[1]
            # Remove 'kvm.' prefix
            if test_name.startswith("kvm."):
                test_name = test_name.split("kvm.")[1]
            result_list.append((test_name, test_status, int(end_time - start_time), info))

        # Found a FAIL/ERROR/GOOD line -- get failure/success info
        elif len(parts) >= 6 and parts[3].startswith("timestamp") and parts[4].startswith("localtime"):
            info_list[-1] = parts[5]

    return result_list


def print_result(result):
    """Nicely print a single Autotest result.

    result -- a 4-tuple
    """
    if result:
        print '%-48s\t\t%s\t%s\t%s' % tuple(map(str, result))


def main(resfile):
    print_result(('test', 'status', 'seconds', 'info'))
    print_result(('----', '------', '-------', '----'))

    f = file(resfile)
    text = f.read()
    f.close()

    results = parse_results(text)
    map(print_result, results)


if __name__ == '__main__':
    import sys, os

    resfile = '../../results/default/status'
    if len(sys.argv) == 2:
        resfile = sys.argv[1]
    if resfile == '-h' or resfile == '--help' or len(sys.argv) > 2:
        print 'usage: %s [ <resfile> ]' % sys.argv[0]
        sys.exit(0)
    if not os.path.exists(resfile):
        print 'Bad result file: %s' % resfile
        sys.exit(1)
    main(resfile)