caizhongang
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
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'):
|
|
# set log
|
|
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)
|
|
|