[logging] adding function for logging
This commit is contained in:
parent
b82e2e6e78
commit
2e86edd639
@ -5,14 +5,28 @@ from rich.logging import RichHandler
|
|||||||
DEFAULT_LOG_LEVEL = "DEBUG"
|
DEFAULT_LOG_LEVEL = "DEBUG"
|
||||||
|
|
||||||
|
|
||||||
def setup_logging(logger: logging.Logger, level=DEFAULT_LOG_LEVEL):
|
def setup_logging(logger: logging.Logger, verbosity: int):
|
||||||
|
level = logging.WARNING # Default level
|
||||||
|
|
||||||
|
if verbosity == 0:
|
||||||
|
level = logging.ERROR
|
||||||
|
elif verbosity == 1:
|
||||||
|
level = logging.INFO
|
||||||
|
elif verbosity >= 2:
|
||||||
|
level = logging.DEBUG
|
||||||
|
print("Debugging enabled.")
|
||||||
|
else:
|
||||||
|
print("Invalid verbosity level. Defaulting to WARNING.")
|
||||||
|
|
||||||
if logger.hasHandlers():
|
if logger.hasHandlers():
|
||||||
logger.handlers.clear()
|
logger.handlers.clear()
|
||||||
stream_handler = RichHandler(rich_tracebacks=True, show_path=False)
|
stream_handler = RichHandler(rich_tracebacks=True, show_path=level == logging.DEBUG)
|
||||||
stream_handler.setLevel(level)
|
stream_handler.setLevel(level)
|
||||||
fmt_shell = "%(filename)s:%(lineno)d - %(message)s"
|
fmt_shell = "%(filename)s:%(lineno)d - %(message)s"
|
||||||
shell_formatter = logging.Formatter(fmt_shell)
|
shell_formatter = logging.Formatter(fmt_shell)
|
||||||
stream_handler.setFormatter(shell_formatter)
|
stream_handler.setFormatter(shell_formatter)
|
||||||
logger.addHandler(stream_handler)
|
logger.addHandler(stream_handler)
|
||||||
logger.setLevel(level)
|
logger.setLevel(level)
|
||||||
|
logger.propagate = False
|
||||||
|
# logging.basicConfig(level=level, handlers=[stream_handler], force=True)
|
||||||
logging.getLogger(__name__).info(f"Logging configured with level {level}.")
|
logging.getLogger(__name__).info(f"Logging configured with level {level}.")
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user