Choose logging handler via --log-handler CLI option (#3293)

This commit is contained in:
Clément Pit-Claudel 2024-05-30 17:13:27 +02:00 committed by GitHub
commit 39c964450c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 22 additions and 3 deletions

View file

@ -126,11 +126,13 @@ logging.setLoggerClass(FatalLogger)
# force root logger to be of our preferred class
logging.getLogger().__class__ = FatalLogger
DEFAULT_LOG_HANDLER = RichHandler(console=console)
def init(
level=None,
fatal="",
handler=RichHandler(console=console),
handler=DEFAULT_LOG_HANDLER,
name=None,
logs_dedup_min_level=None,
):
@ -139,7 +141,10 @@ def init(
LOG_FORMAT = "%(message)s"
logging.basicConfig(
level=level, format=LOG_FORMAT, datefmt="[%H:%M:%S]", handlers=[handler]
level=level,
format=LOG_FORMAT,
datefmt="[%H:%M:%S]",
handlers=[handler] if handler else [],
)
logger = logging.getLogger(name)