Skip to content

Commit f7e5ccc

Browse files
committed
Make log formatting a bit more robust
1 parent 2b2b393 commit f7e5ccc

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

everybody_codes/runner.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616

1717
class LogFormatter(logging.Formatter):
18-
def __init__(self, day: int, *args, **kwargs):
18+
def __init__(self, day: int = 0, *args, **kwargs):
1919
super().__init__(*args, **kwargs)
2020
self.last_call = time.perf_counter_ns()
2121
self.day = day
@@ -25,7 +25,10 @@ def set_part(self, part: int) -> None:
2525
self.part = part
2626

2727
def format(self, record) -> str:
28-
return super().format(record).replace("DAYPART", f"{self.day}.{self.part}")
28+
msg = super().format(record)
29+
if "DAYPART" in msg and self.day and self.part:
30+
msg = msg.replace("DAYPART", f"{self.day}.{self.part}")
31+
return msg
2932

3033
def formatTime(self, record, datefmt=None):
3134
if datefmt:

0 commit comments

Comments
 (0)