Commit 236eb82c authored by Lucas Daniel Velazquez M's avatar Lucas Daniel Velazquez M Committed by GitHub

Adds tqdm handler to logging_config.py for progress bar integration

parent 4afaaf8a
import os import os
import logging import logging
from tqdm.auto import tqdm
class TqdmLoggingHandler(logging.Handler):
def __init__(self, level=logging.INFO):
super().__init__(level)
def emit(self, record):
try:
msg = self.format(record)
tqdm.write(msg)
self.flush()
except Exception:
self.handleError(record)
def setup_logging(loglevel): def setup_logging(loglevel):
if loglevel is None: if loglevel is None:
...@@ -12,5 +25,6 @@ def setup_logging(loglevel): ...@@ -12,5 +25,6 @@ def setup_logging(loglevel):
level=log_level, level=log_level,
format='%(asctime)s %(levelname)s [%(name)s] %(message)s', format='%(asctime)s %(levelname)s [%(name)s] %(message)s',
datefmt='%Y-%m-%d %H:%M:%S', datefmt='%Y-%m-%d %H:%M:%S',
handlers=[TqdmLoggingHandler()]
) )
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment