mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-12-27 07:39:53 +08:00
31 lines
830 B
Python
31 lines
830 B
Python
import os
|
|
import logging
|
|
|
|
from tqdm.auto import tqdm
|
|
|
|
class TqdmLoggingHandler(logging.Handler):
|
|
def __init__(self, level=logging.INFO):
|
|
super().__init__(level)
|
|
|
|
def emit(self, record):
|
|
try:
|
|
msg = self.format(record)
|
|
tqdm.write(msg)
|
|
self.flush()
|
|
except Exception:
|
|
self.handleError(record)
|
|
|
|
def setup_logging(loglevel):
|
|
if loglevel is None:
|
|
loglevel = os.environ.get("SD_WEBUI_LOG_LEVEL")
|
|
|
|
if loglevel:
|
|
log_level = getattr(logging, loglevel.upper(), None) or logging.INFO
|
|
logging.basicConfig(
|
|
level=log_level,
|
|
format='%(asctime)s %(levelname)s [%(name)s] %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S',
|
|
handlers=[TqdmLoggingHandler()]
|
|
)
|
|
|