Improving agent logging

This commit is contained in:
João Moura
2024-03-11 17:05:54 -03:00
parent 1f95d7b982
commit cb5a528550
3 changed files with 13 additions and 6 deletions

View File

@@ -10,7 +10,7 @@ class Logger:
)
self.verbose_level = verbose_level
def log(self, level, message):
def log(self, level, message, color="bold_green"):
level_map = {"debug": 1, "info": 2}
if self.verbose_level and level_map.get(level, 0) <= self.verbose_level:
self._printer.print(f"[{level.upper()}]: {message}", color="bold_green")
self._printer.print(f"[{level.upper()}]: {message}", color=color)