faster Spectrum preview & custom VERBOSE loglvl

1 files changed, 23 insertions, 0 deletions

diff --git a/src/__init__.py b/src/__init__.py
index 2f4cffa..73f174a 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -1,5 +1,28 @@
 import sys
 import os
+import logging
+
+
+class Logger(logging.getLoggerClass()):
+    '''
+        Custom Logger class to handle custom VERBOSE log level.
+        Levels used in this program are as follows:
+            VERBOSE     Annoyingly frequent debug messages (e.g, in loops)
+            DEBUG       Ordinary debug information
+            INFO        Expected events that are expensive or irreversible
+            WARNING     A non-fatal error or suspicious behaviour
+            ERROR       Any error that would interrupt the user
+            CRITICAL    Things that really shouldn't happen at all
+    '''
+    def __init__(self, name, level=logging.NOTSET):
+        super().__init__(name, level)
+        logging.addLevelName(5, "VERBOSE")
+
+    def verbose(self, msg, *args, **kwargs):
+        if self.isEnabledFor(5):
+            self._log(5, msg, args, **kwargs)
+logging.setLoggerClass(Logger)
+logging.VERBOSE = 5
 
 
 if getattr(sys, 'frozen', False):