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

Adds tqdm handler to logging_config.py for progress bar integration

parent 4afaaf8a
Loading
Loading
Loading
Loading
+14 −0
Original line number Diff line number Diff line
import os
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):
    if loglevel is None:
@@ -12,5 +25,6 @@ def setup_logging(loglevel):
            level=log_level,
            format='%(asctime)s %(levelname)s [%(name)s] %(message)s',
            datefmt='%Y-%m-%d %H:%M:%S',
            handlers=[TqdmLoggingHandler()]
        )