diff options
Diffstat (limited to 'src/Logging.py')
| -rw-r--r-- | src/Logging.py | 25 |
1 files changed, 21 insertions, 4 deletions
diff --git a/src/Logging.py b/src/Logging.py index e08160f4..1c33df94 100644 --- a/src/Logging.py +++ b/src/Logging.py @@ -1,5 +1,14 @@ import logging, logging.handlers, os, sys, time +LEVELS = { + "trace": 5, + "debug": logging.DEBUG, + "info": logging.INFO, + "warn": logging.WARN, + "error": logging.ERROR, + "critical": logging.CRITICAL +} + class BitBotFormatter(logging.Formatter): def formatTime(self, record, datefmt=None): ct = self.converter(record.created) @@ -14,9 +23,15 @@ class BitBotFormatter(logging.Formatter): return s class Log(object): - def __init__(self, location): + def __init__(self, level, location): + logging.addLevelName(LEVELS["trace"], "TRACE") self.logger = logging.getLogger(__name__) - self.logger.setLevel(logging.DEBUG) + + if not level.lower() in LEVELS: + raise ValueError("Unknown log level '%s'" % level) + stdout_level = LEVELS[level.lower()] + + self.logger.setLevel(LEVELS["trace"]) formatter = BitBotFormatter( "%(asctime)s [%(levelname)s] %(message)s", @@ -24,16 +39,18 @@ class Log(object): formatter.converter = time.gmtime stdout_handler = logging.StreamHandler(sys.stdout) - stdout_handler.setLevel(logging.INFO) + stdout_handler.setLevel(stdout_level) stdout_handler.setFormatter(formatter) self.logger.addHandler(stdout_handler) file_handler = logging.handlers.TimedRotatingFileHandler( location, when="midnight", backupCount=5) - file_handler.setLevel(logging.DEBUG) + file_handler.setLevel(LEVELS["trace"]) file_handler.setFormatter(formatter) self.logger.addHandler(file_handler) + def trace(self, message, params, **kwargs): + self._log(message, params, LEVELS["trace"], kwargs) def debug(self, message, params, **kwargs): self._log(message, params, logging.DEBUG, kwargs) def info(self, message, params, **kwargs): |