From f7e5ccc692a5401d4aa3856476d18a13a4a7374a Mon Sep 17 00:00:00 2001 From: Isaac Good Date: Sun, 16 Feb 2025 09:09:24 -0800 Subject: [PATCH] Make log formatting a bit more robust --- everybody_codes/runner.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/everybody_codes/runner.py b/everybody_codes/runner.py index 3896131..1d5a578 100755 --- a/everybody_codes/runner.py +++ b/everybody_codes/runner.py @@ -15,7 +15,7 @@ class LogFormatter(logging.Formatter): - def __init__(self, day: int, *args, **kwargs): + def __init__(self, day: int = 0, *args, **kwargs): super().__init__(*args, **kwargs) self.last_call = time.perf_counter_ns() self.day = day @@ -25,7 +25,10 @@ def set_part(self, part: int) -> None: self.part = part def format(self, record) -> str: - return super().format(record).replace("DAYPART", f"{self.day}.{self.part}") + msg = super().format(record) + if "DAYPART" in msg and self.day and self.part: + msg = msg.replace("DAYPART", f"{self.day}.{self.part}") + return msg def formatTime(self, record, datefmt=None): if datefmt: