Add InsLogger

Changes logging solution to INSLogger from the INS team.

Author: wangruoyuyuyu

INSLogger/__init__.py

@@ -0,0 +1,3 @@
+from .logger import Logger
+from .theme import Theme
+from .defaults_theme import ORIGIN_THEME, DEFAULT_THEME

INSLogger/defaults_theme.py

@@ -0,0 +1,19 @@
+from .theme import Theme
+
+ORIGIN_THEME = Theme({
+    "GRAY": "\033[97m",
+    "INFO": "",
+    "WARN": "\033[93m",
+    "ERROR": "\033[91m",
+    "DEBUG": "\033[2m",
+    "CLEAN": "\033[0m"
+})
+
+DEFAULT_THEME = Theme({
+    "GRAY": "\033[97m",
+    "INFO": "\033[36m",
+    "WARN": "\033[33m",
+    "ERROR": "\033[31m",
+    "DEBUG": "\033[34m",
+    "CLEAN": "\033[0m"
+})

INSLogger/logger.py

@@ -0,0 +1,85 @@
+from .theme import Theme
+from .defaults_theme import DEFAULT_THEME
+from typing import TypeVar
+import time
+
+Logger = TypeVar("Logger")
+
+
+class Logger(object):
+    def __init__(self, name: str = "main") -> None:
+        self.name = name
+        self.namespace_gray = False
+        self.disable_time = True
+        self.theme = DEFAULT_THEME
+
+    def config(
+        self,
+        namespace_gray: bool = None,
+        disable_time: bool = None,
+        theme: Theme = None,
+    ) -> None:
+        if namespace_gray is not None:
+            self.namespace_gray = namespace_gray
+        if disable_time is not None:
+            self.disable_time = disable_time
+        if theme is not None:
+            self.theme = theme
+
+    def __call__(self, name: str = "sub") -> Logger:
+        new_logger = Logger(f"{self.name} -> {name}")
+        new_logger.config(self.namespace_gray, self.disable_time, self.theme)
+        return new_logger
+
+    def _log(self, level: str, messages: tuple) -> None:
+        level = level.upper()
+        messages = list(messages)
+
+        msgs = []
+        for i in range(len(messages)):
+            messages[i] = str(messages[i])
+        i = " ".join(messages)
+        for j in i.split("\n"):
+            msgs.append(j)
+
+        for message in msgs:
+            msg = ""
+
+            if self.namespace_gray:
+                msg += self.theme("GRAY")
+            else:
+                msg += self.theme(level)
+            msg += f"[{self.name}] "
+
+            if not self.disable_time:
+                msg += f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] "
+
+            if self.namespace_gray:
+                msg += self.theme(level)
+            msg += f"[{level}] {message} {self.theme('CLEAN')}"
+
+            print(msg)
+
+    def info(self, *message) -> None:
+        self._log("INFO", message)
+
+    def log(self, *message) -> None:
+        self._log("INFO", message)
+
+    def warn(self, *message) -> None:
+        self._log("WARN", message)
+
+    def warning(self, *message) -> None:
+        self._log("WARN", message)
+
+    def error(self, *message) -> None:
+        self._log("ERROR", message)
+
+    def debug(self, *message) -> None:
+        self._log("DEBUG", message)
+
+    @staticmethod
+    def enable() -> None: ...
+
+    @staticmethod
+    def disable() -> None: ...

INSLogger/theme.py

@@ -0,0 +1,49 @@
+from typing import overload, TypeVar
+
+Theme = TypeVar("Theme")
+
+class Theme:
+    @overload
+    def __init__(self) -> None: ...
+    @overload
+    def __init__(self, origin_theme: Theme) -> None: ...
+    @overload
+    def __init__(self, theme_dict: dict) -> None: ...
+
+    def __init__(self, a0 = None) -> None:
+        if isinstance(a0, Theme):
+            self.theme = a0.theme
+        elif isinstance(a0, dict):
+            self.theme = a0
+        else:
+            raise TypeError("a0 only can be Theme or dict!")
+    
+    @overload
+    def set_theme(self, theme_dict: dict) -> None: ...
+    @overload
+    def set_theme(self, theme_key: str, theme_data: str) -> None: ...
+
+    def set_theme(self, a0, a1 = None) -> None:
+        if isinstance(a0, dict):
+            for key in list(a0.keys()):
+                self.theme[key] = a0[key]
+        elif isinstance(a0, str):
+            if a1 is None:
+                raise TypeError("theme_data cannot be None!")
+            if isinstance(a1, str):
+                self.theme[a0] = a1
+            else:
+                raise TypeError("theme_data only can be str!")
+        else:
+            raise TypeError("a0 only can be dict or str!")
+    
+    def get_theme(self, theme_key: str) -> str:
+        if isinstance(theme_key, str):
+            if self.theme.get(theme_key, None) is None:
+                raise KeyError(f"cannot find theme key '{theme_key}'!")
+            return self.theme[theme_key]
+        else:
+            raise TypeError("theme_key only can be str!")
+
+    def __call__(self, theme_key: str) -> str:
+        return self.get_theme(theme_key)