diff options
Diffstat (limited to 'bot/log.py')
| -rw-r--r-- | bot/log.py | 61 |
1 files changed, 32 insertions, 29 deletions
@@ -5,31 +5,28 @@ 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 Logging def setup() -> None: """Set up loggers.""" - # Configure the "TRACE" logging level (e.g. "log.trace(message)") - logging.TRACE = 5 - logging.addLevelName(logging.TRACE, "TRACE") - logging.Logger.trace = _monkeypatch_trace - - format_string = "%(asctime)s | %(name)s | %(levelname)s | %(message)s" - log_format = logging.Formatter(format_string) - root_logger = logging.getLogger() + root_logger = core_logging.get_logger() if Logging.file_logs: # Set up file logging - log_file = Path("logs/sir-lancebot.log") + 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(log_format) + file_handler.setFormatter(core_logging.log_format) root_logger.addHandler(file_handler) if "COLOREDLOGS_LEVEL_STYLES" not in os.environ: @@ -41,32 +38,38 @@ def setup() -> None: } if "COLOREDLOGS_LOG_FORMAT" not in os.environ: - coloredlogs.DEFAULT_LOG_FORMAT = format_string + coloredlogs.DEFAULT_LOG_FORMAT = core_logging.log_format._fmt - coloredlogs.install(level=logging.TRACE, stream=sys.stdout) + coloredlogs.install(level=core_logging.TRACE_LEVEL, stream=sys.stdout) root_logger.setLevel(logging.DEBUG if Logging.debug else logging.INFO) - # Silence irrelevant loggers - logging.getLogger("discord").setLevel(logging.ERROR) - logging.getLogger("websockets").setLevel(logging.ERROR) + logging.getLogger("PIL").setLevel(logging.ERROR) logging.getLogger("matplotlib").setLevel(logging.ERROR) - logging.getLogger("async_rediscache").setLevel(logging.WARNING) _set_trace_loggers() - root_logger.info("Logging initialization complete") -def _monkeypatch_trace(self: logging.Logger, msg: str, *args, **kwargs) -> None: - """ - Log 'msg % args' with severity 'TRACE'. +def setup_sentry() -> None: + """Set up the Sentry logging integrations.""" + sentry_logging = LoggingIntegration( + level=logging.DEBUG, + event_level=logging.WARNING + ) - To pass exception information, use the keyword argument exc_info with a true value, e.g. - logger.trace("Houston, we have an %s", "interesting problem", exc_info=1) - """ - if self.isEnabledFor(logging.TRACE): - self._log(logging.TRACE, msg, args, **kwargs) + sentry_sdk.init( + dsn=os.environ.get("BOT_SENTRY_DSN"), + integrations=[ + sentry_logging, + RedisIntegration(), + ], + release=f"bot@{os.environ.get('GIT_SHA')}", + traces_sample_rate=0.5, + _experiments={ + "profiles_sample_rate": 0.5, + }, + ) def _set_trace_loggers() -> None: @@ -84,13 +87,13 @@ def _set_trace_loggers() -> None: level_filter = Logging.trace_loggers if level_filter: if level_filter.startswith("*"): - logging.getLogger().setLevel(logging.TRACE) + core_logging.get_logger().setLevel(core_logging.TRACE_LEVEL) elif level_filter.startswith("!"): - logging.getLogger().setLevel(logging.TRACE) + core_logging.get_logger().setLevel(core_logging.TRACE_LEVEL) for logger_name in level_filter.strip("!,").split(","): - logging.getLogger(logger_name).setLevel(logging.DEBUG) + core_logging.get_logger(logger_name).setLevel(logging.DEBUG) else: for logger_name in level_filter.strip(",").split(","): - logging.getLogger(logger_name).setLevel(logging.TRACE) + core_logging.get_logger(logger_name).setLevel(core_logging.TRACE_LEVEL) |