63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
#!/usr/bin/python3
|
|
|
|
""" Script that replays curtin events from a journald export.
|
|
curtin events are injected back in journald and log lines are written to a log
|
|
file. """
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import sys
|
|
import time
|
|
from typing import TextIO
|
|
|
|
from systemd import journal
|
|
|
|
scale_factor = float(os.environ.get('SUBIQUITY_REPLAY_TIMESCALE', "4"))
|
|
|
|
def time_for_entry(e):
|
|
return int(e['__MONOTONIC_TIMESTAMP'])/1e6
|
|
|
|
rc = 0
|
|
|
|
def report(e, log_file: TextIO, event_identifier: str):
|
|
global rc
|
|
if e['SYSLOG_IDENTIFIER'].startswith("curtin_event"):
|
|
e['SYSLOG_IDENTIFIER'] = event_identifier
|
|
e['CODE_LINE'] = int(e['CODE_LINE'])
|
|
journal.send(**e)
|
|
r = e.get("CURTIN_RESULT")
|
|
if r == "SUCCESS":
|
|
rc = 0
|
|
elif r == "FAIL":
|
|
rc = 1
|
|
elif e['SYSLOG_IDENTIFIER'].startswith("subiquity_log") and scale_factor < 10:
|
|
log_file.write(e['MESSAGE'] + '\n')
|
|
|
|
|
|
def main() -> int:
|
|
""" Entry point. """
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument("replay-file")
|
|
parser.add_argument("--event-identifier", required=True)
|
|
parser.add_argument("--output", type=argparse.FileType("w"), default="-")
|
|
|
|
args = vars(parser.parse_args())
|
|
|
|
prev_ev = None
|
|
for line in open(args["replay-file"]):
|
|
ev = json.loads(line.strip())
|
|
if prev_ev is not None:
|
|
report(prev_ev, args["output"],
|
|
event_identifier=args["event_identifier"])
|
|
delay = time_for_entry(ev) - time_for_entry(prev_ev)
|
|
time.sleep(min(delay, 8)/scale_factor)
|
|
prev_ev = ev
|
|
report(ev, args["output"], event_identifier=args["event_identifier"])
|
|
return rc
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|