CLI: Added custom logger

pkgs/clan-cli/clan_cli/custom_logger.py

@@ -0,0 +1,42 @@
+import logging
+import datetime
+
+class CustomFormatter(logging.Formatter):
+
+    grey = "\x1b[38;20m"
+    yellow = "\x1b[33;20m"
+    red = "\x1b[31;20m"
+    bold_red = "\x1b[31;1m"
+    green = "\u001b[32m"
+    blue = "\u001b[34m"
+
+    def format_str(color):
+        reset = "\x1b[0m"
+        return f"%(asctime)s - %(name)s - {color} %(levelname)s {reset} - %(message)s (%(filename)s:%(lineno)d)"
+
+    FORMATS = {
+        logging.DEBUG: format_str(blue),
+        logging.INFO: format_str(green),
+        logging.WARNING: format_str(yellow),
+        logging.ERROR: format_str(red),
+        logging.CRITICAL: format_str(bold_red)
+    }
+
+    def formatTime(self, record,datefmt=None):
+        now = datetime.datetime.now()
+        now = now.strftime("%H:%M:%S")
+        return now
+
+    def format(self, record):
+        log_fmt = self.FORMATS.get(record.levelno)
+        formatter = logging.Formatter(log_fmt)
+        formatter.formatTime = self.formatTime
+        return formatter.format(record)
+
+
+def register(level):
+    ch = logging.StreamHandler()
+    ch.setLevel(logging.DEBUG)
+    ch.setFormatter(CustomFormatter())
+    logging.basicConfig(level=level, handlers=[ch])
+