Spaces:
Sleeping
Sleeping
File size: 1,183 Bytes
14d1720 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import logging
import os
import sys
nesting_level = 0
is_start = None
def get_logger(file_name, use_error_log=False, log_dir=None):
logger = logging.getLogger(file_name)
logging_level = getattr(logging, 'INFO')
logger.setLevel(logging_level)
if log_dir is None:
log_dir = os.path.join("./", "log")
if not os.path.exists(log_dir):
os.makedirs(log_dir)
log_dir = os.path.join(log_dir, "log.txt")
else:
log_dir = os.path.join(log_dir, "log.txt")
formatter = logging.Formatter(fmt='%(asctime)s %(filename)s: %(levelname)s: %(message)s')
stdout_handler = logging.StreamHandler(sys.stdout)
stdout_handler.setLevel(logging_level)
stdout_handler.setFormatter(formatter)
logger.addHandler(stdout_handler)
fh = logging.FileHandler(log_dir)
fh.setLevel(logging_level)
fh.setFormatter(formatter)
logger.addHandler(fh)
if use_error_log:
stderr_handler = logging.StreamHandler(sys.stderr)
stderr_handler.setLevel(logging.WARNING)
stderr_handler.setFormatter(formatter)
logger.addHandler(stderr_handler)
logger.propagate = False
return logger
|