diff --git a/generate/log.py b/generate/log.py index a7973b7..6690857 100644 --- a/generate/log.py +++ b/generate/log.py @@ -15,6 +15,7 @@ class Log: messages: list = [] logfile: Path + prefix: str = "" def __init__(self, path: str) -> None: """Init for Log.""" @@ -25,15 +26,19 @@ class Log: def log_error(self, msg: str) -> None: """Log an error.""" now = self.timestamp() - prefix = "EE" - log_message = f"[{now}] ({prefix}) {msg}" + prefix = "(EE)" + if self.prefix != "": + prefix += f" {self.prefix}:" + log_message = f"[{now}] {prefix} {msg}" self.write_message(log_message) def log_warning(self, msg: str) -> None: """Log a warning.""" now = self.timestamp() - prefix = "WW" - log_message = f"[{now}] ({prefix}) {msg}" + prefix = "(WW)" + if self.prefix != "": + prefix += f" {self.prefix}:" + log_message = f"[{now}] {prefix} {msg}" self.write_message(log_message) def write_message(self, msg: str) -> None: @@ -49,3 +54,7 @@ class Log: return datetime.datetime.now(tz=datetime.UTC).strftime( "%Y-%m-%d %H:%M:%S", ) + + def set_prefix(self, prefix: str) -> None: + """Set a prefix.""" + self.prefix = prefix