Coverage for faststream / _internal / logger / formatter.py: 94%

1import logging (empty)

2import sys (empty)

3from typing import Literal (empty)

4 

5COLORED_LEVELS = { (empty)

6 logging.DEBUG: "\033[36mDEBUG\033[0m", 

7 logging.INFO: "\033[32mINFO\033[0m", 

8 logging.WARNING: "\033[33mWARNING\033[0m", 

9 logging.ERROR: "\033[31mERROR\033[0m", 

10 logging.CRITICAL: "\033[91mCRITICAL\033[0m", 

11} 

12 

13 

14class ColourizedFormatter(logging.Formatter): (empty)

15 """A class to format log messages with colorized level names. 

16 

17 Methods: 

18 __init__ : Initialize the formatter with specified format strings. 

19 formatMessage : Format the log record message with colorized level name. 

20 """ 

21 

22 def __init__( (empty)

23 self, 

24 fmt: str | None = None, 

25 datefmt: str | None = None, 

26 style: Literal["%", "{", "$"] = "%", 

27 use_colors: bool | None = None, 

28 ) -> None: 

29 """Initialize the formatter with specified format strings. 

30 

31 Initialize the formatter either with the specified format string, or a 

32 default as described above. Allow for specialized date formatting with 

33 the optional datefmt argument. If datefmt is omitted, you get an 

34 ISO8601-like (or RFC 3339-like) format. 

35 

36 Use a style parameter of '%', '{' or '$' to specify that you want to 

37 use one of %-formatting, :meth:`str.format` (``{}``) formatting or 

38 :class:`string.Template` formatting in your format string. 

39 """ 

40 if use_colors in {True, False}: (empty)

41 self.use_colors = use_colors (empty)

42 else: 

43 self.use_colors = sys.stdout.isatty() (empty)

44 super().__init__(fmt=fmt, datefmt=datefmt, style=style) (empty)

45 

46 def formatMessage(self, record: logging.LogRecord) -> str: # noqa: N802 (empty)

47 """Formats the log message. 

48 

49 Args: 

50 record (logging.LogRecord): The log record to format. 

51 

52 Returns: 

53 str: The formatted log message. 

54 """ 

55 if self.use_colors: 55 ↛ 61line 55 didn't jump to line 61 because the condition on line 55 was always true(empty)

56 record.levelname = expand_log_field( (empty)

57 field=COLORED_LEVELS.get(record.levelno, record.levelname), 

58 symbols=17, 

59 ) 

60 

61 return super().formatMessage(record) (empty)

62 

63 

64def expand_log_field(field: str, symbols: int) -> str: (empty)

65 """Expands a log field by adding spaces. 

66 

67 Args: 

68 field: The log field to expand. 

69 symbols: The desired length of the expanded field. 

70 

71 Returns: 

72 The expanded log field. 

73 """ 

74 return field + (" " * (symbols - len(field))) (empty)