def get_handler(class_name,
level=None,
log_dir=None,
log_type=None,
job_id=None):
if not log_type:
if not LoggerFactory.LOG_DIR or not class_name:
return logging.StreamHandler()
if not log_dir:
log_file = os.path.join(LoggerFactory.LOG_DIR,
"{}.log".format(class_name))
else:
log_file = os.path.join(log_dir, "{}.log".format(class_name))
else:
log_file = os.path.join(
log_dir, "fate_flow_{}.log".format(log_type)
if level == LoggerFactory.LEVEL else
'fate_flow_{}_error.log'.format(log_type))
if job_id:
formatter = logging.Formatter(
LoggerFactory.JOB_LOG_FORMAT.replace("jobid", job_id))
else:
formatter = logging.Formatter(LoggerFactory.LOG_FORMAT)
handler = TimedRotatingFileHandler(log_file,
when='D',
interval=1,
backupCount=14,
delay=True)
if level:
handler.level = level
handler.setFormatter(formatter)
return handler
def get_handler(class_name, level=None, log_dir=None, is_schedule=False):
if not is_schedule:
if not LoggerFactory.LOG_DIR or not class_name:
return logging.StreamHandler()
if not log_dir:
log_file = os.path.join(LoggerFactory.LOG_DIR,
"{}.log".format(class_name))
else:
log_file = os.path.join(log_dir, "{}.log".format(class_name))
else:
log_file = os.path.join(
log_dir, "fate_flow_schedule.log"
if level == LoggerFactory.LEVEL else 'error.log')
formatter = logging.Formatter(
'"%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s"'
)
handler = TimedRotatingFileHandler(log_file,
when='D',
interval=1,
backupCount=14,
delay=True)
if level:
handler.level = level
handler.setFormatter(formatter)
return handler
def get_log_handler(self):
file_handler = TimedRotatingFileHandler(
self.LOG_FILE,
when=self.LOG_WHEN,
interval=self.LOG_INTERVAL,
encoding='UTF-8',
backupCount=self.LOG_FILE_COUNT)
file_handler.suffix = self.LOG_FILE_SUFFIX
file_formatter = logging.Formatter(self.LOG_FORMAT)
file_handler.setFormatter(file_formatter)
file_handler.level = getattr(logging, self.LOG_LEVEL)
return file_handler
def get_handler(level=None, log_dir=None, log_type=None):
log_file = os.path.join(log_dir, "INFO.log" if level == LoggerFactory.LEVEL else 'ERROR.log')
formatter = logging.Formatter(LoggerFactory.FORMAT)
handler = TimedRotatingFileHandler(log_file,
when='D',
interval=1,
backupCount=14,
delay=True)
if level:
handler.level = level
handler.setFormatter(formatter)
return handler
