aboutsummaryrefslogtreecommitdiff
path: root/prettier.py
blob: 1b8ad0aba14c502dda464802ae2f09bbc48ec18f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import os
import sys
import re
import argparse


def green(*strings):
    return "".join([f"\033[32m{s}\033[0m" for s in strings])


def red(*strings):
    return "".join([f"\033[31m{s}\033[0m" for s in strings])

def create_logs_entry(logs, key):
    logs[key] = {}
    logs[key]["ok_counter"] = 0
    logs[key]["ko_counter"] = 0
    logs[key]["ko_info"] = []

def parse():
    logs = {}
    for line in sys.stdin:
        line = line.strip()
        if line[:2] == "OK":
            l = logs.get(line[4:])
            if l is None:
                print("\n\n", line[4:], ": ", sep="", end="")
                create_logs_entry(logs, line[4:])
            logs[line[4:]]["ok_counter"] += 1
            if (logs[line[4:]]["ok_counter"] + logs[line[4:]]["ko_counter"]) % 15 == 0:
                print("\n", ''.join([" " for _ in range(1 + len(line[4:]))]), end="")
            print(green("[OK] "), end="")
            continue
        if line.find("SEGFAULT") != -1:
            name = line[4:]
        else:
            m = re.search("^KO: \[COMPARE\]: (.*): expected: (.*) got: (.*) (with: .*)?$", line)
            if m is None:
                print(line)
                print("PARSING ERROR")
                continue
            name = m.group(1)

        l = logs.get(name)
        if l is None:
            create_logs_entry(logs, name)
            print("\n\n", name, ": ", sep="", end="")
            l = logs.get(name)
        l["ko_counter"] += 1
        l["ko_info"].append({
            "type": "COMPARE",
            "expected": m.group(2),
            "actual": m.group(3),
            "with": m.group(4)[6:]
        })
        if (l["ok_counter"] + l["ko_counter"]) % 15 == 0:
            print("\n", ''.join([" " for _ in range(1 + len(name))]), end="")
        print(red("[KO] "), end="")
        sys.stdout.flush()
    return logs

if __name__ == "__main__":
    os.system("clear")
    logs = parse()
    print("\n")
    for k, v in logs.items():
        for e in v["ko_info"]:
            if e['type'] == "SEGFAULT":
                print(f"{k} : {red('SEGFAULT')}")
            elif e['type'] == "COMPARE":
                print(f"{k} with {e['with']}:\n    {green('expected: ', e['expected'])}\n    {red('actual: ', e['actual'])}")