sd-webui-infinite-image-bro.../scripts/logger.py

21 lines
594 B
Python

import os
from scripts.bin import cwd
import logging
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.INFO)
file_handler = logging.FileHandler(f"{cwd}/log.log")
file_handler.setLevel(logging.DEBUG)
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
console_handler.setFormatter(formatter)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
if "APP_ENV" in os.environ and os.environ["APP_ENV"] == "dev":
logger.addHandler(console_handler)