bitbot-3.11-fork/modules/logging.py

import logging, logging.handlers, sys, time

class BitBotFormatter(logging.Formatter):
    def formatTime(self, record, datefmt=None):
        ct = self.converter(record.created)
        if datefmt:
            if "%f" in datefmt:
                msec = "%03d" % record.msecs
                datefmt = datefmt.replace("%f", msec)
            s = time.strftime(datefmt, ct)
        else:
            t = time.strftime("%Y-%m-%d %H:%M:%S", ct)
            s = "%s.%03d" % (t, record.msecs)
        return s

class Module(object):
    def __init__(self, bot):
        self.logger = logging.getLogger(__name__)
        self.logger.setLevel(logging.DEBUG)

        formatter = BitBotFormatter(
            "%(asctime)s - %(levelname)s - %(message)s",
            "%Y-%m-%dT%H:%M:%S.%f%z")

        stdout_handler = logging.StreamHandler(sys.stdout)
        stdout_handler.setLevel(logging.INFO)
        stdout_handler.setFormatter(formatter)
        self.logger.addHandler(stdout_handler)

        file_handler = logging.handlers.TimedRotatingFileHandler(
            "bot.log", when="midnight", backupCount=5)
        file_handler.setLevel(logging.DEBUG)
        file_handler.setFormatter(formatter)
        self.logger.addHandler(file_handler)

        bot.events.on("log.debug").hook(self.debug)
        bot.events.on("log.info").hook(self.info)
        bot.events.on("log.warn").hook(self.warn)
        bot.events.on("log.error").hook(self.error)
        bot.events.on("log.critical").hook(self.critical)

    def debug(self, event):
        self._log(event, logging.DEBUG)
    def info(self, event):
        self._log(event, logging.INFO)
    def warn(self, event):
        self._log(event, logging.WARN)
    def error(self, event):
        self._log(event, logging.ERROR)
    def critical(self, event):
        self._log(event, logging.CRITICAL)
    def _log(self, event, level):
        message = event["message"]
        params = event.get("params", [])
        self.logger.log(level, message, *params)