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
72
73
74
75
76
77
78
79
|
import re
import sys
from ..reporter_demo import Candidate, Requirement
from .generate import generate_html
from .reporter import GraphGeneratingReporter
def process_arguments(function, args):
function_arg_info = {
"starting": [],
"starting_round": [int],
"ending_round": [int, ...],
"ending": [...],
"adding_requirement": ["requirement", "candidate"],
"backtracking": ["candidate"],
"pinning": ["candidate"],
}
assert function in function_arg_info
retval = []
argument_types = function_arg_info[function]
for arg_type in argument_types:
if arg_type is int:
to_convert, _, args = args.partition(", ")
value = int(to_convert)
elif arg_type == "requirement":
match = re.match(r"^<Requirement\('?([\w\-\._~]+)(.*?)'?\)>(.*)", args)
assert match, repr(args)
name, spec, args = match.groups()
value = Requirement(name, spec)
elif arg_type == "candidate":
match = re.match(r"^(?:<(.+?)==(.+?)>|None)(.*)", args)
assert match, repr(args)
name, version, args = match.groups()
if name and version:
value = Candidate(name, version)
else:
assert not (name or version)
value = None
elif arg_type is ...: # just consume it
value, _, args = args.partition(", ")
else:
raise RuntimeError()
retval.append(value)
if args.startswith(","):
args = args[1:].lstrip()
return function, retval
def parse_line(line):
one = line.strip()
function, _, args = one.partition("(")
assert args[-1] == ")"
args = args[:-1]
return process_arguments(function, args)
def run_reporter_from_logs(reporter, *, logs):
for line in logs:
function_name, args = parse_line(line)
function = getattr(reporter, function_name)
function(*args)
if __name__ == "__main__":
usage = "usage: visualization.py reporter-demo-output.txt out.html"
assert len(sys.argv) == 3, usage
reporter = GraphGeneratingReporter()
with open(sys.argv[1]) as f_one:
run_reporter_from_logs(reporter, logs=f_one)
with open(sys.argv[2], "w") as f_two:
generate_html(reporter.evolution, outfile=f_two)
|
