irma_output_testcase_parser.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
# 2014-11-15 17:33 CET
# y0ug
import sys
import argparse

if __name__ == "__main__":
    data = open(sys.argv[1]).read()
    data = data.split('\n')
    data = data[1:]
    tag = "\tGData Anti-Virus "
    print "    expected_results = {"
    for i in range(0, len(data), 3):
        if len(data[i]) == 0: continue
        filename = data[i]
        sign = data[i+2][len(tag):].strip()
        if sign == "clean":
            print "        '%s': None," % (filename)
        else:
            print "        '%s': '%s'," % (filename, sign)

    print "    }"