Let --verbose have an argument for the log level.

This commit is contained in:
comfyanonymous 2024-10-04 10:05:34 -04:00
parent d854ed0bcf
commit 6f021d8aa0
3 changed files with 4 additions and 4 deletions

View File

@ -10,14 +10,14 @@ def get_logs():
return "\n".join([formatter.format(x) for x in logs])
def setup_logger(verbose: bool = False, capacity: int = 300):
def setup_logger(log_level: str = 'INFO', capacity: int = 300):
global logs
if logs:
return
# Setup default global logger
logger = logging.getLogger()
logger.setLevel(logging.DEBUG if verbose else logging.INFO)
logger.setLevel(log_level)
stream_handler = logging.StreamHandler()
stream_handler.setFormatter(logging.Formatter("%(message)s"))

View File

@ -136,7 +136,7 @@ parser.add_argument("--disable-all-custom-nodes", action="store_true", help="Dis
parser.add_argument("--multi-user", action="store_true", help="Enables per-user storage.")
parser.add_argument("--verbose", action="store_true", help="Enables more debug prints.")
parser.add_argument("--verbose", default='INFO', const='DEBUG', nargs="?", choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'], help='Set the logging level')
# The default built-in provider hosted under web/
DEFAULT_VERSION_STRING = "comfyanonymous/ComfyUI@latest"

View File

@ -9,7 +9,7 @@ from comfy.cli_args import args
from app.logger import setup_logger
setup_logger(verbose=args.verbose)
setup_logger(log_level=args.verbose)
def execute_prestartup_script():