Coverage for src/configuration/custom_formatter.py: 0%

1import logging 

2 

3 

4class CustomFormatter(logging.Formatter): 

5 

6 grey = "\x1b[38;20m" 

7 yellow = "\x1b[33;20m" 

8 red = "\x1b[31;20m" 

9 bold_red = "\x1b[31;1m" 

10 reset = "\x1b[0m" 

11 

12 log_format = "[%(levelname)-5.5s] %(message)s" 

13 

14 FORMATS = { 

15 logging.DEBUG: grey + log_format + reset, 

16 logging.INFO: grey + log_format + reset, 

17 logging.WARNING: yellow + log_format + reset, 

18 logging.ERROR: red + log_format + reset, 

19 logging.CRITICAL: bold_red + log_format + reset 

20 } 

21 

22 def format(self, record): 

23 log_fmt = self.FORMATS.get(record.levelno) 

24 formatter = logging.Formatter(log_fmt) 

25 return formatter.format(record)