From 120acf54eca541b35dbb5501f84704e9681f519b Mon Sep 17 00:00:00 2001
From: jesopo
Date: Mon, 1 Oct 2018 13:48:22 +0100
Subject: Add 'trace' logging level for src/Database.py and src/EventManager.py
 very verbose debugging

---
 src/Logging.py | 25 +++++++++++++++++++++----
 1 file changed, 21 insertions(+), 4 deletions(-)

(limited to 'src/Logging.py')

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):
-- 
cgit v1.3.1-10-gc9f91