Make log formatting a bit more robust

IsaacG · Feb 16, 2025 · f7e5ccc · f7e5ccc
1 parent 2b2b393
commit f7e5ccc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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: