comfyui-discord/lib/log.py
2025-03-04 14:59:10 -07:00

42 lines
1.1 KiB
Python

import sys
import logging
import logging.handlers
LOG_SETUP = False
# Make it easy to log text:
def write(input_str):
global LOG_SETUP
if not LOG_SETUP:
logger = logging.getLogger('discord')
logger.setLevel(logging.DEBUG)
logging.getLogger('discord.http').setLevel(logging.INFO)
logFormatter = logging.Formatter(
"[{asctime}] [{threadName:<16}] [{levelname:<8}] {name}: {message}",
"%Y-%m-%d %H:%M:%S",
style="{")
rootLogger = logging.getLogger()
#fileHandler = logging.FileHandler("log.txt")
fileHandler = logging.handlers.RotatingFileHandler(
filename="log.txt",
encoding="utf-8",
maxBytes=32 * 1024 * 1024,
backupCount=10,
)
fileHandler.setFormatter(logFormatter)
rootLogger.addHandler(fileHandler)
consoleHandler = logging.StreamHandler()
consoleHandler.setFormatter(logFormatter)
rootLogger.addHandler(consoleHandler)
LOG_SETUP = True
logging.getLogger("discord").info(input_str)