from __future__ import print_function
from collections import defaultdict
import xml.etree.ElementTree as ET
import os
import errno
import re

result_file_name='{dir}/results_{tester}_{platform}.txt'
result_info_file_name='{dir}/results_{tester}_{platform}.info'
test_report_filename='{dir}/TestReport_{tester}_{platform}'

xml = open("Test.xml", 'rb').read();

def open_file_create_dir(filename, mode):
    if not os.path.exists(os.path.dirname(filename)):
        try:
            os.makedirs(os.path.dirname(filename))
        except OSError as exc: # Guard against race condition
            if exc.errno != errno.EEXIST:
                raise
    return open(filename, mode)

root=ET.fromstring(xml)
testing = root.find('Testing')
testlist = testing.find('TestList')
nb = 0
tests_ids = {}
for t in testlist:
    tests_ids[t.text] = nb
    nb += 1

tests = {}
labels = set()
for t in testing.findall('Test'):
    tests[tests_ids[t.find('FullName').text]] = \
         { \
           "Name":   t.find('Name').text, \
           "Status": t.attrib['Status'], \
           "Output": t.find('Results').find('Measurement').find('Value').text, \
           "Labels": [l.text for l in t.find('Labels').findall('Label')] if t.find('Labels') is not None else ['UNKNOWN_LABEL'], \
         }

print("""
1
2
3
4
5
6
7
8
9
10
11
12
""", file=open(result_info_file_name.format(dir=os.getcwd(),
                                            tester="me",
                                            platform="Mon Linux"), 'w'))

tests_per_label = defaultdict(list)
for t_id in range(0, len(tests)):
    t = tests[t_id]
    for label in t['Labels']:
        labels.add(label)
        tests_per_label[label].append(t)

with open_file_create_dir(result_file_name.format(dir=os.getcwd(),
                                                  tester="me",
                                                  platform="Mon Linux"), 'w') as results:
    for label, tests in tests_per_label.items():
        result_for_label='y'
        with open_file_create_dir("{}/error.txt".format(label), 'w') as error:
            for t in tests:
                print("   {} {}".format("successful " if (t['Status'] == 'passed') else "ERROR:     ", t['Name']), file=error)
                if t['Status'] != 'passed':
                    result_for_label='n'
                elif t['Output'] != None and re.match(r'(^|[^a-zA-Z_,:-])warning', t['Output']):
                    result_for_label='y'

                with open("{}/ProgramOutput.{}".format(label, t['Name']), 'w') as f:
                    f.write(t['Output'] if t['Output'] != None else "")


            print("{label} {result}".format(label=label, result=result_for_label), file=results)


for label, tests in tests_per_label.items():
        with open_file_create_dir(test_report_filename.format(dir=label,
                                                              tester="me",
                                                              platform="Mon Linux"), 'w') as label_report:
            print("""
------------------------------------------------------------------
- Error output from platform {platform}
------------------------------------------------------------------

{error_txt}
"""               .format(platform="Mon Linux",
                         error_txt=open("{}/error.txt".format(label), 'r').read()), file=label_report)
            for t in tests:
                filename="{}/ProgramOutput.{}".format(label, t['Name'])
                with open(filename, 'r') as f:
                    print("""
------------------------------------------------------------------
- {file}
------------------------------------------------------------------

{file_text}
"""               .format(file=filename,
                          file_text=f.read()), file=label_report)