import logging
from logging import handlers
class Logger(object):
level_relations = {
'debug':logging.DEBUG,
'info':logging.INFO,
'warning':logging.WARNING,
'error':logging.ERROR,
'crit':logging.CRITICAL
}#日志级别关系映射
def __init__(self,filename,level='info',fmt='%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s'):
self.logger = logging.getLogger(filename)
format_str = logging.Formatter(fmt)#设置日志格式
self.logger.setLevel(self.level_relations.get(level))#设置日志级别
sh = logging.StreamHandler()#往屏幕上输出
sh.setFormatter(format_str) #设置屏幕上显示的格式
th = handlers.TimedRotatingFileHandler(filename = filename, encoding='utf-8')
th.setFormatter(format_str)#设置文件里写入的格式
self.logger.addHandler(sh) #代表在屏幕上输出,如果注释掉,屏幕将不输出
self.logger.addHandler(th) #代表在log文件中输出,如果注释掉,将不再向文件中写入数据
if __name__ == '__main__':
log = Logger('all.log',level='debug')
log.logger.debug('debug')
log.logger.info('info')
log.logger.warning('警告')
log.logger.error('报错')
log.logger.critical('严重')
Logger('error.log', level='error').logger.error('error')
可以直接将main函数上面的代码粘贴进文件中,使用import对其进行调用,实现输出内容既可以写到文件中,也可以输出到屏幕上,也可以利用self.logger.addHandler(sh) #代表在屏幕上输出,如果注释掉,屏幕将不输出 self.logger.addHandler(th) #代表在log文件中输出,如果注释掉,将不再向文件中写入数据
控制是否输出在屏幕或者是否输出到文件。
下面是代码运行截图:
屏幕截图:
log文件截图:
可以明显看出,屏幕输出和log文件中写入的内容是一样的!