colog

colors and logging

modules/log.py

@@ -0,0 +1,59 @@
+import logging
+from datetime import datetime
+from modules.settings import *
+
+class CustomFormatter(logging.Formatter):
+    grey = '\x1b[38;21m'
+    white = '\x1b[38;5;231m'
+    blue = '\x1b[38;5;39m'
+    yellow = '\x1b[38;5;226m'
+    red = '\x1b[38;5;196m'
+    green = '\x1b[38;5;46m'
+    purple = '\x1b[38;5;129m'
+    bold_red = '\x1b[31;1m'
+    bold_white = '\x1b[37;1m'
+    reset = '\x1b[0m'
+
+    def __init__(self, fmt):
+        super().__init__()
+        self.fmt = fmt
+        self.FORMATS = {
+            logging.DEBUG: self.blue + self.fmt + self.reset,
+            logging.INFO: self.white + self.fmt + self.reset,
+            logging.WARNING: self.yellow + self.fmt + self.reset,
+            logging.ERROR: self.red + self.fmt + self.reset,
+            logging.CRITICAL: self.bold_red + self.fmt + self.reset
+        }
+
+    def format(self, record):
+        log_fmt = self.FORMATS.get(record.levelno)
+        formatter = logging.Formatter(log_fmt)
+        return formatter.format(record)
+
+# Create logger
+logger = logging.getLogger("MeshBot System Logger")
+logger.setLevel(logging.DEBUG)
+logger.propagate = False
+
+msgLogger = logging.getLogger("MeshBot Messages Logger")
+msgLogger.setLevel(logging.INFO)
+msgLogger.propagate = False
+
+# Define format for logs
+logFormat = '%(asctime)s | %(levelname)8s | %(message)s'
+msgLogFormat = '%(asctime)s | %(message)s'
+
+# Create stdout handler for logging to the console
+stdout_handler = logging.StreamHandler()
+stdout_handler.setLevel(logging.DEBUG)
+stdout_handler.setFormatter(CustomFormatter(logFormat))
+
+# Create file handler for logging to a file (logs INFO level and above)
+today = datetime.now()
+file_handler = logging.FileHandler('messages{}.log'.format(today.strftime('%Y_%m_%d')))
+file_handler.setLevel(logging.INFO)
+file_handler.setFormatter(logging.Formatter(msgLogFormat))
+
+# Add handlers to the logger
+logger.addHandler(stdout_handler)
+msgLogger.addHandler(file_handler)