Improving endpoint

pkgs/clan-cli/clan_cli/custom_logger.py

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