Skip to content

Commit 61d75f5

Browse files
committed
Introduce combine_logs.py to combine log files from multiple bitcoinds.
This commit adds a tool for combining log files from multiple instances of bitcoinds as well as the test_framework.log file. This gives a combined view of what the test framework and all bitcoin instances were doing during a qa test.
1 parent 919aaf6 commit 61d75f5

File tree

2 files changed

+151
-0
lines changed

2 files changed

+151
-0
lines changed

test/functional/combine_logs.py

+111
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,111 @@
1+
#!/usr/bin/env python3
2+
"""Combine logs from multiple bitcoin nodes as well as the test_framework log.
3+
4+
This streams the combined log output to stdout. Use combine_logs.py > outputfile
5+
to write to an outputfile."""
6+
7+
import argparse
8+
from collections import defaultdict, namedtuple
9+
import glob
10+
import heapq
11+
import os
12+
import re
13+
import sys
14+
15+
# Matches on the date format at the start of the log event
16+
TIMESTAMP_PATTERN = re.compile(r"^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{6}")
17+
18+
LogEvent = namedtuple('LogEvent', ['timestamp', 'source', 'event'])
19+
20+
def main():
21+
"""Main function. Parses args, reads the log files and renders them as text or html."""
22+
23+
parser = argparse.ArgumentParser(usage='%(prog)s [options] <test temporary directory>', description=__doc__)
24+
parser.add_argument('-c', '--color', dest='color', action='store_true', help='outputs the combined log with events colored by source (requires posix terminal colors. Use less -r for viewing)')
25+
parser.add_argument('--html', dest='html', action='store_true', help='outputs the combined log as html. Requires jinja2. pip install jinja2')
26+
args, unknown_args = parser.parse_known_args()
27+
28+
if args.color and os.name != 'posix':
29+
print("Color output requires posix terminal colors.")
30+
sys.exit(1)
31+
32+
if args.html and args.color:
33+
print("Only one out of --color or --html should be specified")
34+
sys.exit(1)
35+
36+
# There should only be one unknown argument - the path of the temporary test directory
37+
if len(unknown_args) != 1:
38+
print("Unexpected arguments" + str(unknown_args))
39+
sys.exit(1)
40+
41+
log_events = read_logs(unknown_args[0])
42+
43+
print_logs(log_events, color=args.color, html=args.html)
44+
45+
def read_logs(tmp_dir):
46+
"""Reads log files.
47+
48+
Delegates to generator function get_log_events() to provide individual log events
49+
for each of the input log files."""
50+
51+
files = [("test", "%s/test_framework.log" % tmp_dir)]
52+
for i, logfile in enumerate(glob.glob("%s/node*/regtest/debug.log" % tmp_dir)):
53+
files.append(("node%d" % i, logfile))
54+
55+
return heapq.merge(*[get_log_events(source, f) for source, f in files])
56+
57+
def get_log_events(source, logfile):
58+
"""Generator function that returns individual log events.
59+
60+
Log events may be split over multiple lines. We use the timestamp
61+
regex match as the marker for a new log event."""
62+
try:
63+
with open(logfile, 'r') as infile:
64+
event = ''
65+
timestamp = ''
66+
for line in infile:
67+
# skip blank lines
68+
if line == '\n':
69+
continue
70+
# if this line has a timestamp, it's the start of a new log event.
71+
time_match = TIMESTAMP_PATTERN.match(line)
72+
if time_match:
73+
if event:
74+
yield LogEvent(timestamp=timestamp, source=source, event=event.rstrip())
75+
event = line
76+
timestamp = time_match.group()
77+
# if it doesn't have a timestamp, it's a continuation line of the previous log.
78+
else:
79+
event += "\n" + line
80+
# Flush the final event
81+
yield LogEvent(timestamp=timestamp, source=source, event=event.rstrip())
82+
except FileNotFoundError:
83+
print("File %s could not be opened. Continuing without it." % logfile, file=sys.stderr)
84+
85+
def print_logs(log_events, color=False, html=False):
86+
"""Renders the iterator of log events into text or html."""
87+
if not html:
88+
colors = defaultdict(lambda: '')
89+
if color:
90+
colors["test"] = "\033[0;36m" # CYAN
91+
colors["node0"] = "\033[0;34m" # BLUE
92+
colors["node1"] = "\033[0;32m" # GREEN
93+
colors["node2"] = "\033[0;31m" # RED
94+
colors["node3"] = "\033[0;33m" # YELLOW
95+
colors["reset"] = "\033[0m" # Reset font color
96+
97+
for event in log_events:
98+
print("{0} {1: <5} {2} {3}".format(colors[event.source.rstrip()], event.source, event.event, colors["reset"]))
99+
100+
else:
101+
try:
102+
import jinja2
103+
except ImportError:
104+
print("jinja2 not found. Try `pip install jinja2`")
105+
sys.exit(1)
106+
print(jinja2.Environment(loader=jinja2.FileSystemLoader('./'))
107+
.get_template('combined_log_template.html')
108+
.render(title="Combined Logs from testcase", log_events=[event._asdict() for event in log_events]))
109+
110+
if __name__ == '__main__':
111+
main()
+40
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
<html lang="en">
2+
<head>
3+
<title> {{ title }} </title>
4+
<style>
5+
ul {
6+
list-style-type: none;
7+
font-family: monospace;
8+
}
9+
li {
10+
border: 1px solid slategray;
11+
margin-bottom: 1px;
12+
}
13+
li:hover {
14+
filter: brightness(85%);
15+
}
16+
li.log-test {
17+
background-color: cyan;
18+
}
19+
li.log-node0 {
20+
background-color: lightblue;
21+
}
22+
li.log-node1 {
23+
background-color: lightgreen;
24+
}
25+
li.log-node2 {
26+
background-color: lightsalmon;
27+
}
28+
li.log-node3 {
29+
background-color: lightyellow;
30+
}
31+
</style>
32+
</head>
33+
<body>
34+
<ul>
35+
{% for event in log_events %}
36+
<li class="log-{{ event.source }}"> {{ event.source }} {{ event.timestamp }} {{event.event}}</li>
37+
{% endfor %}
38+
</ul>
39+
</body>
40+
</html>

0 commit comments

Comments
 (0)