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
|
import difflib
import os
import re
from collections import defaultdict
from pathlib import Path
from check.focs_dump_parser._tools import get_base_dump_folder, get_branch
_negative_number = re.compile(r"\(-\(?\d+\.?\d*\)\)")
def read_file(path: Path):
if not path.exists():
return []
else:
with path.open() as f:
return f.readlines()
def prettify_dump(text):
text = text.replace("(0 = LocalCandidate.TurnsSinceFocusChange)", "(LocalCandidate.TurnsSinceFocusChange = 0)")
def replacer(match):
return match.group(0).replace("(", "").replace(")", "")
return _negative_number.sub(replacer, text)
def report_diff():
base = get_base_dump_folder()
a_branch = "master"
b_branch = get_branch()
path_a = base / a_branch
path_b = base / b_branch
files_a = os.listdir(path_a)
files_b = os.listdir(path_b)
all_files = {*files_a, *files_b}
clean_species = []
stats = defaultdict(list)
for name in all_files:
content_a = read_file(path_a / name)
content_b = read_file(path_b / name)
name = name.removesuffix(".txt")
a = [prettify_dump(x) for x in content_a]
b = [prettify_dump(x) for x in content_b]
if a == b == []:
stats["both empty"].append(name)
if a == b:
stats["same"].append(name)
elif content_a == []:
stats["file mismatch"].append(f"{name}.{a_branch} is missed")
elif content_b == []:
stats["file mismatch"].append(f"{name}.{b_branch} is missed")
else:
stats["diff"].append(name)
for line in difflib.unified_diff(a, b, f"{name}.{a_branch}", f"{name}.{b_branch}"):
print(line, end="")
print(clean_species)
print(f"Finish comparing {len(all_files)} files")
return stats
if __name__ == "__main__":
report_diff()
|
