121 lines
5.0 KiB
Python
121 lines
5.0 KiB
Python
# BotLibrary/library/logger.py
|
||
# Создание логгеров и подключение функций логирования в файлы
|
||
|
||
import sys
|
||
from loguru import logger
|
||
|
||
from BotLibrary.configs import ProjectPath
|
||
from BotLibrary.analitics.message_to_file import write_message_to_file
|
||
from BotLibrary.analitics.user_data_to_file import write_user_info_to_file
|
||
|
||
# Настройка экспорта модулей и логирования
|
||
__all__ = ("logger", "setup_logger", "cmd_logginger", "error_cmd_logginger",
|
||
"logginger", "common_msg_logginger", )
|
||
|
||
|
||
# Класс для параметров логгера
|
||
class LogsSet:
|
||
# Максимальный размер лог-файла
|
||
max_size = "500 MB"
|
||
|
||
# Шаблон логов для информации
|
||
info_text = (
|
||
"<green>{time:YYYY-MM-DD HH:mm:ss}</green> <red>|</red> "
|
||
"<blue>PRIMO-{extra[log_type]}</blue> <red>|</red> "
|
||
"<red>{extra[user]} |</red> <level>{message}</level>"
|
||
)
|
||
|
||
# Шаблон логов для ошибок
|
||
error_text = (
|
||
"<level>{time:YYYY-MM-DD HH:mm:ss} | "
|
||
"<bold>ERROR-{extra[log_type]}</bold> | "
|
||
"{extra[user]} | {message}</level>"
|
||
)
|
||
|
||
# Шаблон логов для отладки
|
||
debug_text = (
|
||
"<cyan>{time:YYYY-MM-DD HH:mm:ss}</cyan> <red>|</red> "
|
||
"<magenta>DEBUG-{extra[log_type]}</magenta> <red>|</red> "
|
||
"<yellow>{extra[user]} |</yellow> <level>{message}</level>"
|
||
)
|
||
|
||
# Шаблон логов для предупреждений
|
||
warning_text = (
|
||
"<yellow>{time:YYYY-MM-DD HH:mm:ss}</yellow> <red>|</red> "
|
||
"<orange>WARNING-{extra[log_type]}</orange> <red>|</red> "
|
||
"<red>{extra[user]} |</red> <level>{message}</level>"
|
||
)
|
||
|
||
|
||
# Создание обычного логгера + логгер в файл
|
||
async def setup_logger():
|
||
logger.remove() # Удаляем все логгеры
|
||
|
||
# Пустой логгер для записи отступов в файл уровня TRACE
|
||
logger.add(ProjectPath.log_file,
|
||
rotation=LogsSet.max_size,
|
||
format="\n\n\n",
|
||
backtrace=True,
|
||
diagnose=True,
|
||
level="TRACE",
|
||
filter=lambda record: record["level"].name == "TRACE")
|
||
logger.trace("")
|
||
|
||
# Логгер для вывода в консоль и файл с уровнем INFO
|
||
logger.add(sys.stderr,
|
||
colorize=True,
|
||
format=LogsSet.info_text,
|
||
level="INFO",
|
||
filter=lambda record: record["level"].name == "INFO")
|
||
logger.add(ProjectPath.log_file,
|
||
rotation=LogsSet.max_size,
|
||
format=LogsSet.info_text,
|
||
backtrace=True,
|
||
diagnose=True,
|
||
level="INFO",
|
||
filter=lambda record: record["level"].name == "INFO")
|
||
|
||
# Логгер для вывода в консоль и файл с уровнем ERROR
|
||
logger.add(sys.stderr,
|
||
colorize=True,
|
||
format=LogsSet.error_text,
|
||
level="ERROR",
|
||
filter=lambda record: record["level"].name == "ERROR")
|
||
logger.add(ProjectPath.log_error_file,
|
||
rotation=LogsSet.max_size,
|
||
format=LogsSet.error_text,
|
||
backtrace=True,
|
||
diagnose=True,
|
||
level="ERROR",
|
||
filter=lambda record: record["level"].name == "ERROR")
|
||
|
||
|
||
# Запись сообщения в файл и информации о пользователи
|
||
async def logginger(message):
|
||
write_user_info_to_file(message.from_user)
|
||
await write_message_to_file(message)
|
||
|
||
|
||
# Создание функции логирования на обычные сообщения
|
||
async def common_msg_logginger(message, name, message_type, log_type):
|
||
# Проверка на наличие текста и его типа
|
||
if message.text is None:
|
||
logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(
|
||
f"Получено сообщение из ({name}) : {message_type}")
|
||
else:
|
||
logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(
|
||
f"Получено сообщение из ({name}) : {message.text}")
|
||
|
||
|
||
# Специальный логгер для команд. Вывод в консоль, файл и запись информации о пользователи
|
||
async def cmd_logginger(message, log_type, text):
|
||
await logginger(message)
|
||
logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(text)
|
||
|
||
|
||
# Специальный логгер для ошибок с командами. Вывод в консоль, файл и запись информации о пользователи
|
||
async def error_cmd_logginger(message, log_type, e):
|
||
text_error = f"Ошибка при использовании команды /{log_type.lower()}: {str(e)}\n"
|
||
await logginger(message)
|
||
logger.bind(log_type=log_type, user=f"@{message.from_user.username}").error(text_error)
|