From ffeb61bda04cb0aa135d6508938854810a271e07 Mon Sep 17 00:00:00 2001
From: lulzette <ivan@lulzette.ru>
Date: Fri, 2 Jun 2023 00:10:24 +0300
Subject: [PATCH] =?UTF-8?q?[#11]=20=D0=97=D0=B0=D0=BC=D0=B5=D0=BD=D0=B8?=
 =?UTF-8?q?=D0=BB=20colored=20=D0=BD=D0=B0=20=D0=BD=D0=B0=D1=82=D0=B8?=
 =?UTF-8?q?=D0=B2=D0=BD=D1=8B=D0=B5=20=D1=86=D0=B2=D0=B5=D1=82=D0=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 daemon.py | 53 ++++++++++++++++++++++++++++++++++++++---------------
 1 file changed, 38 insertions(+), 15 deletions(-)

diff --git a/daemon.py b/daemon.py
index 86f0254..b5aa3a2 100644
--- a/daemon.py
+++ b/daemon.py
@@ -12,7 +12,6 @@ import subprocess
 import time
 import logging
 from logging.handlers import TimedRotatingFileHandler
-from termcolor import colored
 
 log_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s')
 log_file = 'output.log'
@@ -27,10 +26,10 @@ def set_config():
     config = configparser.ConfigParser()
     if not config.read('cfg_file.ini'):
         config["app"] = {
-                "path": "",
-                "check_period": 5,
-                "max_files": 3
-            }
+            "path": "",
+            "check_period": 5,
+            "max_files": 3
+        }
         config["twitch"] = {
             "app_id": "",
             "app_secret": "",
@@ -108,10 +107,7 @@ def check_stream():
         resolved_id = twitch_client.get_users(logins=[i])
         if not resolved_id['data']:
             log.error(
-                colored(
-                    "Аккаунт " + i + " не найден",
-                    'red',
-                )
+                "Аккаунт " + i + " не найден"
             )
             continue
         # Создаем путь до диры со стримером, если папка не существует
@@ -131,11 +127,7 @@ def check_stream():
                 open(config['app']['path'] + "/pid", 'w').close
             else:
                 log.info(
-                    colored(
-                        "Идет запись " + i,
-                        'red',
-                        attrs=['bold']
-                    )
+                    "Идет запись " + i
                 )
         else:
             # Если стрим не идет, то пишем об этом и убираем его из залоченных
@@ -177,10 +169,41 @@ def get_file_handler():
     return file_handler
 
 
+class CustomFormatter(logging.Formatter):
+
+    grey = "\x1b[38;20m"
+    yellow = "\x1b[33;20m"
+    red = "\x1b[31;20m"
+    bold_red = "\x1b[31;1m"
+    reset = "\x1b[0m"
+    format = "%(asctime)s %(levelname)s - %(message)s"
+
+    FORMATS = {
+        logging.DEBUG: grey + format + reset,
+        logging.INFO: grey + format + reset,
+        logging.WARNING: yellow + format + reset,
+        logging.ERROR: red + format + reset,
+        logging.CRITICAL: bold_red + format + reset
+    }
+
+    def format(self, record):
+        log_fmt = self.FORMATS.get(record.levelno)
+        formatter = logging.Formatter(log_fmt)
+        return formatter.format(record)
+
+
 def get_logger(logger_name):
+    """
+    Инициализация лога
+    """
     logger = logging.getLogger(logger_name)
     logger.setLevel(logging.DEBUG)
-    logger.addHandler(get_console_handler())
+
+    # Console logging
+    console = get_console_handler()
+    console.setFormatter(CustomFormatter())
+    logger.addHandler(console)
+
     logger.addHandler(get_file_handler())
     logger.propagate = False
     return logger