ereuse/devicehub-django
8
0
Fork 0
Code Issues 6 Pull requests 6 Packages Projects Releases Wiki Activity

utils/logger: ensure msgs are logged

Browse source
This commit is contained in:
pedro 2024-11-27 00:52:20 +01:00
parent e25b8e5994
commit 09f6cd2e68
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
utils/logger.py
View file

@ -22,8 +22,14 @@ class CustomFormatter(logging.Formatter):
record.levelname = f"{color}{record.levelname}{RESET}" record.levelname = f"{color}{record.levelname}{RESET}"
if record.args: if record.args:
record.msg = self.highlight_args(record.msg, record.args, color) try:
record.args = () record.msg = record.msg % record.args
record.args = ()
except (TypeError, ValueError):
record.msg = f"{color}{record.msg}{RESET}"
# Highlight the final formatted message
record.msg = self.highlight_message(record.msg, color)
# provide trace when DEBUG config # provide trace when DEBUG config
if settings.DEBUG: if settings.DEBUG:
@ -33,8 +39,4 @@ class CustomFormatter(logging.Formatter):
return super().format(record) return super().format(record)
def highlight_args(self, message, args, color): def highlight_args(self, message, args, color):
try: return f"{color}{message}{RESET}"
highlighted_args = tuple(f"{color}{arg}{RESET}" for arg in args)
return message % highlighted_args
except (TypeError, ValueError):
return f"{color}{message}{RESET}"