Update aiogram 3 template

tgbot/utils/misc/bot_logging.py

@@ -1,32 +1,58 @@
 # - *- coding: utf- 8 - *-
-import logging as bot_logger
+import logging
+from logging.handlers import RotatingFileHandler
+from pathlib import Path
 
 import colorlog
 
 from tgbot.data.config import PATH_LOGS
 
-# Logging format
-log_formatter_file = bot_logger.Formatter("%(levelname)s | %(asctime)s | %(filename)s:%(lineno)d | %(message)s")
-log_formatter_console = colorlog.ColoredFormatter(
-    "%(purple)s%(levelname)s %(blue)s|%(purple)s %(asctime)s %(blue)s|%(purple)s %(filename)s:%(lineno)d %(blue)s|%(purple)s %(message)s%(red)s",
-    datefmt="%d-%m-%Y %H:%M:%S",
-)
+LOG_FILE_MAX_BYTES = 5 * 1024 * 1024
+LOG_FILE_BACKUP_COUNT = 5
 
-# Logging in file logs.log
-file_handler = bot_logger.FileHandler(PATH_LOGS, "w", "utf-8")
-file_handler.setFormatter(log_formatter_file)
-file_handler.setLevel(bot_logger.INFO)
+# Папка под логи создается сама, чтобы бот не падал на старте
+log_path = Path(PATH_LOGS)
+log_path.parent.mkdir(parents=True, exist_ok=True)
 
-# Logging in console
-console_handler = bot_logger.StreamHandler()
-console_handler.setFormatter(log_formatter_console)
-console_handler.setLevel(bot_logger.CRITICAL)
+# Один общий логгер для всего шаблона
+bot_logger = logging.getLogger("tgbot")
+bot_logger.setLevel(logging.INFO)
+bot_logger.propagate = False
 
-# Connect logging settings
-bot_logger.basicConfig(
-    format="%(levelname)s | %(asctime)s | %(filename)s:%(lineno)d | %(message)s",
-    handlers=[
-        file_handler,
-        console_handler
-    ]
-)
+if not bot_logger.handlers:
+    # Формат для файла: без цветов
+    file_formatter = logging.Formatter(
+        "%(levelname)s | %(asctime)s | %(name)s | %(filename)s:%(lineno)d | %(message)s",
+        datefmt="%d-%m-%Y %H:%M:%S",
+    )
+    # Формат для консоли: коротко и с цветами
+    console_formatter = colorlog.ColoredFormatter(
+        "%(log_color)s%(levelname)s%(reset)s | %(blue)s%(asctime)s%(reset)s | "
+        "%(purple)s%(filename)s:%(lineno)d%(reset)s | %(message)s",
+        datefmt="%d-%m-%Y %H:%M:%S",
+        log_colors={
+            "DEBUG": "cyan",
+            "INFO": "green",
+            "WARNING": "yellow",
+            "ERROR": "red",
+            "CRITICAL": "bold_red",
+        },
+    )
+
+    # Не очищаем файл, а крутим по размеру
+    file_handler = RotatingFileHandler(
+        log_path,
+        maxBytes=LOG_FILE_MAX_BYTES,
+        backupCount=LOG_FILE_BACKUP_COUNT,
+        encoding="utf-8",
+    )
+    file_handler.setFormatter(file_formatter)
+    file_handler.setLevel(logging.INFO)
+
+    # В консоль выводим то, что важно видеть сразу
+    console_handler = logging.StreamHandler()
+    console_handler.setFormatter(console_formatter)
+    console_handler.setLevel(logging.INFO)
+
+    bot_logger.addHandler(file_handler)
+    bot_logger.addHandler(console_handler)