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]) 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 global logs
if logs: if logs:
return return
# Setup default global logger # Setup default global logger
logger = logging.getLogger() logger = logging.getLogger()
logger.setLevel(logging.DEBUG if verbose else logging.INFO) logger.setLevel(log_level)
stream_handler = logging.StreamHandler() stream_handler = logging.StreamHandler()
stream_handler.setFormatter(logging.Formatter("%(message)s")) 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("--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/ # The default built-in provider hosted under web/
DEFAULT_VERSION_STRING = "comfyanonymous/ComfyUI@latest" DEFAULT_VERSION_STRING = "comfyanonymous/ComfyUI@latest"

View File

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