Skip to content

log

dev_tool.log

logger

A context manager for setting up and tearing down logging.

This context manager configures file and console logging handlers.

:yield: None

Source code in dev_tool/log.py 
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
@contextmanager
def logger() -> Generator[None, Any, None]:
    """
    A context manager for setting up and tearing down logging.

    This context manager configures file and console logging handlers.

    :yield: None
    """

    try:
        log = logging.getLogger()
        log.setLevel(logging.DEBUG)

        loggers = [
            'httpx',
            'httpcore.connection',
            'httpcore.http11',
            'httpcore.http2',
            'urllib3'
        ]

        for logger_name in loggers:
            logging.getLogger(logger_name).setLevel(logging.WARNING)

        path = STRATUS / 'stratus.log'
        path.touch(exist_ok=True)

        file_handler = RotatingFileHandler(
            filename=path,
            encoding='utf-8',
            maxBytes=10 * 1024 * 1024,
            backupCount=5
        )

        stream_handler = logging.StreamHandler()

        color_formatter = colorlog.ColoredFormatter(
            fmt='{log_color}[{asctime}] [{levelname}] {name}: {message}',
            datefmt='%Y-%m-%d %I:%M:%S %p',
            style='{',
            log_colors={
                'DEBUG': 'cyan',
                'INFO': 'green',
                'WARNING': 'yellow',
                'ERROR': 'red',
                'CRITICAL': 'bold_red',
            },
        )

        formatter = logging.Formatter(
            '[{asctime}] [{levelname}] {name}: {message}',
            '%Y-%m-%d %I:%M:%S %p',
            style='{',
        )

        file_handler.setFormatter(formatter)
        file_handler.setLevel(logging.DEBUG)

        stream_handler.setFormatter(color_formatter)
        stream_handler.setLevel(logging.CRITICAL)

        log.addHandler(file_handler)
        log.addHandler(stream_handler)

        log.propagate = False

        yield
    finally:
        for handler in log.handlers:
            handler.close()
            log.removeHandler(handler)