bril2json 0.1.0

Convert textual Bril to JSON
Documentation 
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 sys, subprocess, json
import multiprocessing
import difflib


def init_worker(shared_failure_event, shared_oracle):
    global failure_event
    global oracle
    failure_event = shared_failure_event
    oracle = shared_oracle


def check_file(file):
    oracle_output = json.loads(subprocess.getoutput(f"{oracle} <{file}"))
    my_output = json.loads(subprocess.getoutput(f"cargo run --quiet -- {file}"))
    oracle_pretty = subprocess.run(
        "bril2txt",
        input=json.dumps(oracle_output).encode("utf-8"),
        stdout=subprocess.PIPE,
    ).stdout.decode("utf-8")
    my_pretty = subprocess.run(
        "bril2txt",
        input=json.dumps(my_output).encode("utf-8"),
        stdout=subprocess.PIPE,
    ).stdout.decode("utf-8")
    if oracle_pretty == my_pretty:
        print(f"\x1b[32m{file} OK\x1b[m")
    else:
        print(f"\x1b[31m{file} ERROR\x1b[m")
        failure_event.set()

        red = lambda text: f"\033[38;2;255;0;0m{text}\033[m"
        green = lambda text: f"\033[38;2;0;255;0m{text}\033[m"
        blue = lambda text: f"\033[38;2;0;0;255m{text}\033[m"
        white = lambda text: f"\033[38;2;255;255;255m{text}\033[m"
        gray = lambda text: f"\033[38;2;128;128;128m{text}\033[m"

        diff = difflib.ndiff(oracle_pretty.splitlines(), my_pretty.splitlines())
        print("--- DIFF ---")
        for line in diff:
            if line.startswith("+"):
                print(green(line))
            elif line.startswith("-"):
                print(red(line))
            elif line.startswith("^"):
                print(blue(line))
            elif line.startswith("?"):
                print(gray(line))
            else:
                print(white(line))


if __name__ == "__main__":
    oracle = sys.argv[1]
    files = sys.argv[2:]

    with multiprocessing.Manager() as manager:
        failure_event = manager.Event()

        with multiprocessing.Pool(
            multiprocessing.cpu_count(),
            initializer=init_worker,
            initargs=(failure_event, oracle),
        ) as pool:
            pool.imap_unordered(check_file, files)
            pool.close()
            pool.join()
            if failure_event.is_set():
                print("Exiting due to errors")
                pool.terminate()
                sys.exit(1)