fix: 修复日志重叠的问题

amiyabot/log/handlers.py

@@ -2,7 +2,7 @@
 import sys
 import logging
 
-from typing import Optional
+from typing import Optional, Dict
 from concurrent_log_handler import ConcurrentRotatingFileHandler
 from amiyabot.util import argv
 
@@ -17,25 +17,34 @@ class LogHandlers:
     level = logging.DEBUG if debug_mode else logging.INFO
     formatter = logging.Formatter(f'%(asctime)s [%(name)9s][%(levelname)9s]%(message)s')
     stream_handler: Optional[logging.StreamHandler] = None
+    file_handlers: Dict[str, ConcurrentRotatingFileHandler] = {}
 
     @classmethod
-    def get_stream_handler(cls):
+    def set_stream_handler(cls, logger: logging.Logger):
         if not cls.stream_handler:
             cls.stream_handler = logging.StreamHandler(stream=sys.stdout)
             cls.stream_handler.setFormatter(cls.formatter)
             cls.stream_handler.setLevel(cls.level)
 
-        return cls.stream_handler
+        if cls.stream_handler not in logger.handlers:
+            logger.addHandler(cls.stream_handler)
 
     @classmethod
-    def get_file_handler(cls, save_path: str, save_filename: str):
-        file_handler = ConcurrentRotatingFileHandler(
-            filename=os.path.join(save_path, f'{save_filename}.log'),
-            encoding='utf-8',
-            maxBytes=LOG_FILE_MAX_BYTES,
-            backupCount=LOG_FILE_BACKUP_COUNT,
-        )
-        file_handler.setFormatter(cls.formatter)
-        file_handler.setLevel(cls.level)
-
-        return file_handler
+    def set_file_handler(cls, logger: logging.Logger, save_path: str, save_filename: str):
+        file_path = os.path.join(save_path, f'{save_filename}.log')
+        if file_path not in cls.file_handlers:
+            file_handler = ConcurrentRotatingFileHandler(
+                filename=file_path,
+                encoding='utf-8',
+                maxBytes=LOG_FILE_MAX_BYTES,
+                backupCount=LOG_FILE_BACKUP_COUNT,
+            )
+            file_handler.setFormatter(cls.formatter)
+            file_handler.setLevel(cls.level)
+
+            cls.file_handlers[file_path] = file_handler
+        else:
+            file_handler = cls.file_handlers[file_path]
+
+        if file_handler not in logger.handlers:
+            logger.addHandler(file_handler)

amiyabot/log/manager.py

@@ -17,8 +17,9 @@ def __init__(self, name: str, save_path: str = LOG_FILE_SAVE_PATH, save_filename
 
         self.__logger = logging.getLogger(name=name)
         self.__logger.setLevel(LogHandlers.level)
-        self.__logger.addHandler(LogHandlers.get_stream_handler())
-        self.__logger.addHandler(LogHandlers.get_file_handler(save_path, save_filename))
+
+        LogHandlers.set_stream_handler(self.__logger)
+        LogHandlers.set_file_handler(self.__logger, save_path, save_filename)
 
     @classmethod
     def use(cls, logger_cls):