You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.4 KiB
56 lines
1.4 KiB
#!/usr/bin/env python
|
|
# coding: utf8
|
|
|
|
""" Centralized logging facilities for Spleeter. """
|
|
|
|
import logging
|
|
import warnings
|
|
from os import environ
|
|
|
|
# pyright: reportMissingImports=false
|
|
# pylint: disable=import-error
|
|
from typer import echo
|
|
|
|
# pylint: enable=import-error
|
|
|
|
__email__ = "spleeter@deezer.com"
|
|
__author__ = "Deezer Research"
|
|
__license__ = "MIT License"
|
|
|
|
environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
|
|
|
|
|
class TyperLoggerHandler(logging.Handler):
|
|
""" A custom logger handler that use Typer echo. """
|
|
|
|
def emit(self, record: logging.LogRecord) -> None:
|
|
echo(self.format(record))
|
|
|
|
|
|
formatter = logging.Formatter("%(levelname)s:%(name)s:%(message)s")
|
|
handler = TyperLoggerHandler()
|
|
handler.setFormatter(formatter)
|
|
logger: logging.Logger = logging.getLogger("spleeter")
|
|
logger.addHandler(handler)
|
|
logger.setLevel(logging.INFO)
|
|
|
|
|
|
def configure_logger(verbose: bool) -> None:
|
|
"""
|
|
Configure application logger.
|
|
|
|
Parameters:
|
|
verbose (bool):
|
|
`True` to use verbose logger, `False` otherwise.
|
|
"""
|
|
from tensorflow import get_logger
|
|
from tensorflow.compat.v1 import logging as tf_logging
|
|
|
|
tf_logger = get_logger()
|
|
tf_logger.handlers = [handler]
|
|
if verbose:
|
|
tf_logging.set_verbosity(tf_logging.INFO)
|
|
logger.setLevel(logging.DEBUG)
|
|
else:
|
|
warnings.filterwarnings("ignore")
|
|
tf_logging.set_verbosity(tf_logging.ERROR)
|