Sfoglia il codice sorgente

Logging: set formatter correctly for fallback logger too

Aarni Koskela 1 anno fa
parent
commit
0011640ab1
1 ha cambiato i file con 6 aggiunte e 5 eliminazioni
  1. 6 5
      modules/logging_config.py

+ 6 - 5
modules/logging_config.py

@@ -36,20 +36,21 @@ def setup_logging(loglevel):
         # Already configured, do not interfere
         return
 
+    formatter = logging.Formatter(
+        '%(asctime)s %(levelname)s [%(name)s] %(message)s',
+        '%Y-%m-%d %H:%M:%S',
+    )
+
     if os.environ.get("SD_WEBUI_RICH_LOG"):
         from rich.logging import RichHandler
         handler = RichHandler()
     else:
         handler = logging.StreamHandler()
+        handler.setFormatter(formatter)
 
     if TqdmLoggingHandler:
         handler = TqdmLoggingHandler(handler)
 
-    formatter = logging.Formatter(
-        '%(asctime)s %(levelname)s [%(name)s] %(message)s',
-        '%Y-%m-%d %H:%M:%S',
-    )
-
     handler.setFormatter(formatter)
 
     log_level = getattr(logging, loglevel.upper(), None) or logging.INFO