1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
"""Wrappers around the logging module."""
import functools
import logging
import sys
import typing
from logging import (
CRITICAL,
DEBUG,
ERROR,
FATAL,
INFO,
NOTSET,
StreamHandler,
WARN,
WARNING,
getLogger,
root,
)
import colorlog.formatter
__all__ = (
"CRITICAL",
"DEBUG",
"ERROR",
"FATAL",
"INFO",
"NOTSET",
"WARN",
"WARNING",
"StreamHandler",
"basicConfig",
"critical",
"debug",
"error",
"exception",
"getLogger",
"info",
"log",
"root",
"warning",
)
def basicConfig(
style: colorlog.formatter._FormatStyle = "%",
log_colors: typing.Optional[colorlog.formatter.LogColors] = None,
reset: bool = True,
secondary_log_colors: typing.Optional[colorlog.formatter.SecondaryLogColors] = None,
format: str = "%(log_color)s%(levelname)s%(reset)s:%(name)s:%(message)s",
datefmt: typing.Optional[str] = None,
**kwargs
) -> None:
"""Call ``logging.basicConfig`` and override the formatter it creates."""
logging.basicConfig(**kwargs)
def _basicConfig():
handler = logging.root.handlers[0]
handler.setFormatter(
colorlog.formatter.ColoredFormatter(
fmt=format,
datefmt=datefmt,
style=style,
log_colors=log_colors,
reset=reset,
secondary_log_colors=secondary_log_colors,
stream=kwargs.get("stream", None),
)
)
if sys.version_info >= (3, 13):
with logging._lock: # type: ignore
_basicConfig()
else:
logging._acquireLock() # type: ignore
try:
_basicConfig()
finally:
logging._releaseLock() # type: ignore
def ensure_configured(func):
"""Modify a function to call our basicConfig() first if no handlers exist."""
@functools.wraps(func)
def wrapper(*args, **kwargs):
if len(logging.root.handlers) == 0:
basicConfig()
return func(*args, **kwargs)
return wrapper
debug = ensure_configured(logging.debug)
info = ensure_configured(logging.info)
warning = ensure_configured(logging.warning)
error = ensure_configured(logging.error)
critical = ensure_configured(logging.critical)
log = ensure_configured(logging.log)
exception = ensure_configured(logging.exception)
|
