updated logger and TODO-shki

modules/logger.py

@@ -0,0 +1,48 @@
+import logging
+import sys
+
+class ColoredFormatter(logging.Formatter):
+    COLORS = {  # ANSI escape sequences for colors
+        'DEBUG': '\033[35m',    # Purple
+        'INFO': '\033[36m',     # Green
+        'WARNING': '\033[33m',   # Yellow
+        'ERROR': '\033[31m',     # Red
+        'CRITICAL': '\033[41m',  # Red background
+    }
+    RESET = '\033[0m'  # Reset to default color
+
+    def format(self, record):
+        color = self.COLORS.get(record.levelname, self.RESET)
+        message = super().format(record)
+        formatted_message = f"{color}{message}{self.RESET}"
+        if record.levelname == 'CRITICAL':
+            print(formatted_message)
+            sys.exit(1)
+        return formatted_message
+
+
+log = logging.getLogger() # Настраиваем логирование
+def set_logging_level(level_string="INFO"):
+    levels = {
+        'DEBUG': logging.DEBUG,
+        'INFO': logging.INFO,
+        'WARNING': logging.WARNING,
+        'ERROR': logging.ERROR,
+        'CRITICAL': logging.CRITICAL
+    }
+
+    level = levels.get(level_string, logging.WARNING)  # Уровень по умолчанию
+    if level == logging.WARNING:
+        print(f"Неизвестный уровень логирования: {level_string}. Устанавливается уровень WARNING.")
+    return level
+
+handler = logging.StreamHandler()  # Создаем консольный обработчик
+handler.setFormatter(ColoredFormatter('%(levelname)s: %(message)s'))
+
+# TODO: добавить условие для управления переменой пути сохранения лога
+log_file_path = 'portproton.log'  # Это может быть переменная, установленная пользователем
+file_handler = logging.FileHandler(log_file_path)  # Создаем файловый обработчик
+file_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s: %(message)s'))
+
+log.addHandler(file_handler)
+log.addHandler(handler)