| import logging |
| import os |
|
|
| OK = '\033[92m' |
| WARNING = '\033[93m' |
| FAIL = '\033[91m' |
| END = '\033[0m' |
|
|
| PINK = '\033[95m' |
| BLUE = '\033[94m' |
| GREEN = OK |
| RED = FAIL |
| WHITE = END |
| YELLOW = WARNING |
|
|
| class colorlogger(): |
| def __init__(self, log_dir, log_name='train_logs.txt'): |
| |
| self._logger = logging.getLogger(log_name) |
| self._logger.setLevel(logging.INFO) |
| log_file = os.path.join(log_dir, log_name) |
| if not os.path.exists(log_dir): |
| os.makedirs(log_dir) |
| file_log = logging.FileHandler(log_file, mode='a') |
| file_log.setLevel(logging.INFO) |
| console_log = logging.StreamHandler() |
| console_log.setLevel(logging.INFO) |
| formatter = logging.Formatter( |
| "{}%(asctime)s{} %(message)s".format(GREEN, END), |
| "%m-%d %H:%M:%S") |
| file_log.setFormatter(formatter) |
| console_log.setFormatter(formatter) |
| self._logger.addHandler(file_log) |
| self._logger.addHandler(console_log) |
|
|
| def debug(self, msg): |
| self._logger.debug(str(msg)) |
|
|
| def info(self, msg): |
| self._logger.info(str(msg)) |
|
|
| def warning(self, msg): |
| self._logger.warning(WARNING + 'WRN: ' + str(msg) + END) |
|
|
| def critical(self, msg): |
| self._logger.critical(RED + 'CRI: ' + str(msg) + END) |
|
|
| def error(self, msg): |
| self._logger.error(RED + 'ERR: ' + str(msg) + END) |
|
|
|
|
