pymss 1.0__py3-none-any.whl

pymss/__init__.py

@@ -0,0 +1,2 @@
+from .separator import MSSeparator
+from .logger import get_separation_logger

pymss/logger.py

@@ -0,0 +1,93 @@
+import logging
+import os
+from colorama import Fore, Style, init
+from datetime import datetime
+
+init(autoreset=True)
+
+MAX_LOG = 100
+LOG_DIR = "logs"
+
+class ColorFormatter(logging.Formatter):
+    def format(self, record):
+        record.pathname = os.path.relpath(record.pathname)
+        log_msg = super().format(record)
+
+        if record.levelname == "INFO":
+            log_msg = log_msg.replace("INFO", f"{Fore.GREEN}INFO{Style.RESET_ALL}")
+        elif record.levelname == "DEBUG":
+            log_msg = log_msg.replace("DEBUG", f"{Fore.BLUE}DEBUG{Style.RESET_ALL}")
+        elif record.levelname == "WARNING":
+            log_msg = log_msg.replace("WARNING", f"{Fore.YELLOW}WARNING{Style.RESET_ALL}")
+        elif record.levelname == "ERROR":
+            log_msg = log_msg.replace("ERROR", f"{Fore.RED}ERROR{Style.RESET_ALL}")
+        elif record.levelname == "CRITICAL":
+            log_msg = log_msg.replace("CRITICAL", f"{Fore.MAGENTA}CRITICAL{Style.RESET_ALL}")
+
+        return log_msg
+
+
+def manage_log_files(log_dir, max_log):
+    log_files = [f for f in os.listdir(log_dir) if f.endswith(".log")]
+
+    def parse_date(filename):
+        for fmt in ("%Y-%m-%d", "%Y-%m-%d_%H-%M-%S"):
+            try:
+                return datetime.strptime(filename.split(".")[0], fmt)
+            except ValueError:
+                continue
+        return datetime.min
+
+    log_files = sorted(log_files, key=parse_date)
+
+    while len(log_files) > max_log:
+        try:
+            oldest_file = log_files.pop(0)
+            os.remove(os.path.join(log_dir, oldest_file))
+        except: pass
+
+
+def set_log_level(logger, level):
+    logger.console_handler.setLevel(level)
+
+
+def get_separation_logger(console_level=logging.INFO, enable_file_log=False, max_log=MAX_LOG):
+    logger = logging.getLogger("logger")
+
+    if logger.hasHandlers():
+        return logger
+
+    logger.setLevel(logging.DEBUG)
+
+    console_handler = logging.StreamHandler()
+    console_handler.setLevel(console_level)
+    formatter = ColorFormatter(fmt="%(asctime)s.%(msecs)03d [%(levelname)s] [%(pathname)s:%(lineno)d] %(message)s", datefmt="%H:%M:%S")
+    console_handler.setFormatter(formatter)
+
+    logger.addHandler(console_handler)
+
+    if enable_file_log:
+        os.makedirs(LOG_DIR, exist_ok=True)
+        log_filename = datetime.now().strftime("%Y-%m-%d_%H-%M-%S.log")
+        file_path = os.path.join(LOG_DIR, log_filename)
+
+        file_handler = logging.FileHandler(file_path, mode='a', encoding='utf-8')
+        file_handler.setLevel(logging.DEBUG)
+        file_formatter = logging.Formatter(fmt="%(asctime)s.%(msecs)03d [%(levelname)s] [%(pathname)s:%(lineno)d] %(message)s", datefmt="%H:%M:%S")
+        file_handler.setFormatter(file_formatter)
+        
+        logger.addHandler(file_handler)
+        manage_log_files(LOG_DIR, max_log)
+
+    logger.console_handler = console_handler
+
+    return logger
+
+
+if __name__ == "__main__":
+    logger = get_separation_logger(console_level=logging.INFO, enable_file_log=False)
+    logger.debug("This is a debug message.")
+    logger.info("This is an info message.")
+    logger.warning("This is a warning message.")
+    logger.error("This is an error message.")
+    logger.critical("This is a critical message.")