[#11] Заменил colored на нативные цвета

This commit is contained in:
lulzette 2023-06-02 00:10:24 +03:00
parent bc72da118e
commit ffeb61bda0

View File

@ -12,7 +12,6 @@ import subprocess
import time import time
import logging import logging
from logging.handlers import TimedRotatingFileHandler from logging.handlers import TimedRotatingFileHandler
from termcolor import colored
log_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s') log_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s')
log_file = 'output.log' log_file = 'output.log'
@ -27,10 +26,10 @@ def set_config():
config = configparser.ConfigParser() config = configparser.ConfigParser()
if not config.read('cfg_file.ini'): if not config.read('cfg_file.ini'):
config["app"] = { config["app"] = {
"path": "", "path": "",
"check_period": 5, "check_period": 5,
"max_files": 3 "max_files": 3
} }
config["twitch"] = { config["twitch"] = {
"app_id": "", "app_id": "",
"app_secret": "", "app_secret": "",
@ -108,10 +107,7 @@ def check_stream():
resolved_id = twitch_client.get_users(logins=[i]) resolved_id = twitch_client.get_users(logins=[i])
if not resolved_id['data']: if not resolved_id['data']:
log.error( log.error(
colored( "Аккаунт " + i + " не найден"
"Аккаунт " + i + " не найден",
'red',
)
) )
continue continue
# Создаем путь до диры со стримером, если папка не существует # Создаем путь до диры со стримером, если папка не существует
@ -131,11 +127,7 @@ def check_stream():
open(config['app']['path'] + "/pid", 'w').close open(config['app']['path'] + "/pid", 'w').close
else: else:
log.info( log.info(
colored( "Идет запись " + i
"Идет запись " + i,
'red',
attrs=['bold']
)
) )
else: else:
# Если стрим не идет, то пишем об этом и убираем его из залоченных # Если стрим не идет, то пишем об этом и убираем его из залоченных
@ -177,10 +169,41 @@ def get_file_handler():
return 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): def get_logger(logger_name):
"""
Инициализация лога
"""
logger = logging.getLogger(logger_name) logger = logging.getLogger(logger_name)
logger.setLevel(logging.DEBUG) 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.addHandler(get_file_handler())
logger.propagate = False logger.propagate = False
return logger return logger