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
|
import logging
import logging.handlers
import os
import sys
from pathlib import Path
import coloredlogs
import sentry_sdk
from pydis_core.utils import logging as core_logging
from sentry_sdk.integrations.logging import LoggingIntegration
from sentry_sdk.integrations.redis import RedisIntegration
from bot.constants import Client, GIT_SHA, Logging
def setup() -> None:
"""Set up loggers."""
root_logger = core_logging.get_logger()
if Logging.file_logs:
# Set up file logging
log_file = Path("logs", "sir-lancebot.log")
log_file.parent.mkdir(exist_ok=True)
# File handler rotates logs every 5 MB
file_handler = logging.handlers.RotatingFileHandler(
log_file, maxBytes=5 * (2 ** 20), backupCount=10, encoding="utf-8",
)
file_handler.setFormatter(core_logging.log_format)
root_logger.addHandler(file_handler)
if "COLOREDLOGS_LEVEL_STYLES" not in os.environ:
coloredlogs.DEFAULT_LEVEL_STYLES = {
**coloredlogs.DEFAULT_LEVEL_STYLES,
"trace": {"color": 246},
"critical": {"background": "red"},
"debug": coloredlogs.DEFAULT_LEVEL_STYLES["info"],
}
if "COLOREDLOGS_LOG_FORMAT" not in os.environ:
coloredlogs.DEFAULT_LOG_FORMAT = core_logging.log_format._fmt
coloredlogs.install(level=core_logging.TRACE_LEVEL, stream=sys.stdout)
root_logger.setLevel(logging.DEBUG if Logging.debug else logging.INFO)
logging.getLogger("PIL").setLevel(logging.ERROR)
logging.getLogger("matplotlib").setLevel(logging.ERROR)
_set_trace_loggers()
root_logger.info("Logging initialization complete")
def setup_sentry() -> None:
"""Set up the Sentry logging integrations."""
sentry_logging = LoggingIntegration(
level=logging.DEBUG,
event_level=logging.WARNING
)
sentry_sdk.init(
dsn=Client.sentry_dsn,
integrations=[
sentry_logging,
RedisIntegration(),
],
release=f"bot@{GIT_SHA}",
traces_sample_rate=0.5,
_experiments={
"profiles_sample_rate": 0.5,
},
)
def _set_trace_loggers() -> None:
"""
Set loggers to the trace level according to the value from the BOT_TRACE_LOGGERS env var.
When the env var is a list of logger names delimited by a comma,
each of the listed loggers will be set to the trace level.
If this list is prefixed with a "!", all of the loggers except the listed ones will be set to the trace level.
Otherwise if the env var begins with a "*",
the root logger is set to the trace level and other contents are ignored.
"""
level_filter = Logging.trace_loggers
if level_filter:
if level_filter.startswith("*"):
core_logging.get_logger().setLevel(core_logging.TRACE_LEVEL)
elif level_filter.startswith("!"):
core_logging.get_logger().setLevel(core_logging.TRACE_LEVEL)
for logger_name in level_filter.strip("!,").split(","):
core_logging.get_logger(logger_name).setLevel(logging.DEBUG)
else:
for logger_name in level_filter.strip(",").split(","):
core_logging.get_logger(logger_name).setLevel(core_logging.TRACE_LEVEL)
|
