python之log模块封装

# CreateTime: 2019/5/8 20:03 # Author: li jianhua import logging import time import os class Log(object): def __init__(self, log_dir): """ 指定保存日志的文件路径，日志级别，以及调用文件将日志存入到指定的文件中 """ self.log_dir = log_dir if not os.path.exists(self.log_dir): os.mkdir(self.log_dir) def get_log(self, log_name, mode='a'): # 创建一个logger logger_name = log_name[:-4] if log_name.endswith('.log') else log_name logger = logging.getLogger(logger_name) logger.setLevel(logging.DEBUG) # 创建一个handler，用于写入日志文件 log_path = os.path.join(self.log_dir, log_name) fh = logging.FileHandler(log_path, mode=mode, encoding='utf-8') fh.setLevel(logging.INFO) # 创建一个handler，用于输出到控制台 ch = logging.StreamHandler() ch.setLevel(logging.INFO) # 定义handler的输出格式 formatter = logging.Formatter( '[%(asctime)s] %(filename)s->%(funcName)s line:%(lineno)d [%(levelname)s]%(message)s') fh.setFormatter(formatter) ch.setFormatter(formatter) # 给logger添加handler logger.addHandler(fh) logger.addHandler(ch) # 关闭打开的文件 fh.close() ch.close() return logger log_dir = r'/home/gzblue/ljh/virtual_proj/smoke/region/chengdu/output_data/log_data' base_log = Log(log_dir) check_config_logger = base_log.get_log('check_config.log', mode='w') background_preprocess_logger = base_log.get_log('background_preprocess.log', mode='w') main_logger = base_log.get_log('main.log', mode='a') if __name__ == '__main__': log_dir = r'log_data' log_name = r'test.log' log = Log(log_dir) logger = log.get_log(log_name) logger.info('This is a') logger.info('This is b') logger.warning('This is c')