103 lines
2.8 KiB
Python
103 lines
2.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
-------------------------------------------------
|
|||
|
|
File Name： LogHandler.py
|
|||
|
|
Description : 日志操作模块
|
|||
|
|
Author : JHao
|
|||
|
|
date： 2017/3/6
|
|||
|
|
-------------------------------------------------
|
|||
|
|
Change Activity:
|
|||
|
|
2017/3/6: log handler
|
|||
|
|
2017/9/21: 屏幕输出/文件输出 可选(默认屏幕和文件均输出)
|
|||
|
|
-------------------------------------------------
|
|||
|
|
"""
|
|||
|
|
__author__ = 'JHao'
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
from logging.handlers import TimedRotatingFileHandler
|
|||
|
|
|
|||
|
|
# 日志级别
|
|||
|
|
CRITICAL = 50
|
|||
|
|
FATAL = CRITICAL
|
|||
|
|
ERROR = 40
|
|||
|
|
WARNING = 30
|
|||
|
|
WARN = WARNING
|
|||
|
|
INFO = 20
|
|||
|
|
DEBUG = 10
|
|||
|
|
NOTSET = 0
|
|||
|
|
|
|||
|
|
CURRENT_PATH = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
ROOT_PATH = os.path.join(CURRENT_PATH, os.pardir)
|
|||
|
|
LOG_PATH = os.path.join(ROOT_PATH, 'log')
|
|||
|
|
|
|||
|
|
if not os.path.exists(LOG_PATH):
|
|||
|
|
os.mkdir(LOG_PATH)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LogHandler(logging.Logger):
|
|||
|
|
"""
|
|||
|
|
LogHandler
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(self, name, level=DEBUG, stream=True, file=True):
|
|||
|
|
self.name = name
|
|||
|
|
self.level = level
|
|||
|
|
logging.Logger.__init__(self, self.name, level=level)
|
|||
|
|
if stream:
|
|||
|
|
self.__setStreamHandler__()
|
|||
|
|
if file:
|
|||
|
|
self.__setFileHandler__()
|
|||
|
|
|
|||
|
|
def __setFileHandler__(self, level=None):
|
|||
|
|
"""
|
|||
|
|
set file handler
|
|||
|
|
:param level:
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
file_name = os.path.join(LOG_PATH, '{name}.log'.format(name=self.name))
|
|||
|
|
# 设置日志回滚, 保存在log目录, 一天保存一个文件, 保留15天
|
|||
|
|
file_handler = TimedRotatingFileHandler(filename=file_name, when='D', interval=1, backupCount=15)
|
|||
|
|
file_handler.suffix = '%Y%m%d.log'
|
|||
|
|
if not level:
|
|||
|
|
file_handler.setLevel(self.level)
|
|||
|
|
else:
|
|||
|
|
file_handler.setLevel(level)
|
|||
|
|
formatter = logging.Formatter('%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s')
|
|||
|
|
|
|||
|
|
file_handler.setFormatter(formatter)
|
|||
|
|
self.file_handler = file_handler
|
|||
|
|
self.addHandler(file_handler)
|
|||
|
|
|
|||
|
|
def __setStreamHandler__(self, level=None):
|
|||
|
|
"""
|
|||
|
|
set stream handler
|
|||
|
|
:param level:
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
stream_handler = logging.StreamHandler()
|
|||
|
|
formatter = logging.Formatter('%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s')
|
|||
|
|
stream_handler.setFormatter(formatter)
|
|||
|
|
if not level:
|
|||
|
|
stream_handler.setLevel(self.level)
|
|||
|
|
else:
|
|||
|
|
stream_handler.setLevel(level)
|
|||
|
|
self.addHandler(stream_handler)
|
|||
|
|
|
|||
|
|
def resetName(self, name):
|
|||
|
|
"""
|
|||
|
|
reset name
|
|||
|
|
:param name:
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
self.name = name
|
|||
|
|
self.removeHandler(self.file_handler)
|
|||
|
|
self.__setFileHandler__()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
log = LogHandler('test')
|
|||
|
|
log.info('this is a test msg')
|