55 lines
2 KiB
Python
55 lines
2 KiB
Python
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)
|