Spaces:
Sleeping
Sleeping
| import logging | |
| import tqdm | |
| import utils.file as uf | |
| def get_logger(name, log_path, isMain=False, level=logging.INFO): | |
| formatter = logging.Formatter( | |
| fmt="%(asctime)s: %(module)s.%(funcName)s +%(lineno)s: %(levelname)-8s %(message)s", | |
| datefmt="%H:%M:%S" | |
| ) | |
| logger = logging.getLogger(name) | |
| logger.setLevel(level) | |
| # # Logging to console | |
| stream_handler = logging.StreamHandler() | |
| stream_handler.setFormatter(formatter) | |
| logger.addHandler(stream_handler) | |
| # Logging to a file | |
| if isMain: | |
| uf.make_directory(log_path, is_dir=False) | |
| file_handler = logging.FileHandler(log_path) | |
| file_handler.setFormatter(formatter) | |
| logger.addHandler(file_handler) | |
| return logger | |
| def progress_bar(iterable, total, **kwargs): | |
| return tqdm.tqdm(iterable=iterable, total=total, ascii=True, **kwargs) | |