# coding=utf8
"""
@author bfzs
"""
import os
import logging
from logging.handlers import TimedRotatingFileHandler, RotatingFileHandler
from cloghandler import ConcurrentRotatingFileHandlerformat_dict = {1: logging.Formatter('%(asctime)s - %(name)s - %(lineno)d - %(levelname)s -  %(message)s', "%Y-%m-%d %H:%M:%S"),2: logging.Formatter('%(asctime)s - %(name)s - %(lineno)d - %(levelname)s -  %(message)s', "%Y-%m-%d %H:%M:%S"),3: logging.Formatter('%(asctime)s - %(name)s - %(lineno)d - %(levelname)s -  %(message)s', "%Y-%m-%d %H:%M:%S"),4: logging.Formatter('%(asctime)s - %(name)s - %(lineno)d - %(levelname)s -  %(message)s', "%Y-%m-%d %H:%M:%S"),5: logging.Formatter('%(asctime)s - %(name)s - %(lineno)d - %(levelname)s -  %(message)s', "%Y-%m-%d %H:%M:%S"),
}class LogLevelException(Exception):def __init__(self, log_level):err = '设置的日志级别是 {0}， 设置错误，请设置为1 2 3 4 5 范围的数字'.format(log_level)Exception.__init__(self, err)class Log(object):"""一个日志类，用于创建和捕获日志，支持将日志打印到控制台打印和写入日志文件。"""def __init__(self, logger_name=None, log_level_int=1, is_add_stream_handler=True, log_path=None, log_filename=None):""":param logger_name: 日志名称，当为None时候打印所有日志:param log_level_int: 日志级别，设置为 1 2 3 4 5，分别对应DEBUG，INFO，WARNING，ERROR,CRITICAL:param is_add_stream_handler: 是否打印日志到控制台:param log_path: 设置存放日志的文件夹路径:param log_filename: 日志的名字，仅当log_path和log_filename都不为None时候才写入到日志文件。:type logger_name :str:type log_level_int :int:type is_add_stream_handler :bool:type log_path :str:type log_filename :str"""self.logger = logging.getLogger(logger_name)self.__check_log_level(log_level_int)self._logger_level = self.__transform_logger_level(log_level_int)self._is_add_stream_handler = is_add_stream_handlerself._log_path = log_pathself._log_filename = log_filenameself._formatter = format_dict[log_level_int]self.__set_logger_level()self.__add_handlers()def debug(self, msg):self.logger.debug(msg)def info(self, msg):self.logger.info(msg)def warning(self, msg):self.logger.warning(msg)def error(self, msg):self.logger.error(msg)def critical(self, msg):self.logger.critical(msg)def __set_logger_level(self):self.logger.setLevel(self._logger_level)@staticmethoddef __check_log_level(log_level_int):if log_level_int not in [1, 2, 3, 4, 5]:raise LogLevelException(log_level_int)@staticmethoddef __transform_logger_level(log_level_int):logger_level = Noneif log_level_int == 1:logger_level = logging.DEBUGelif log_level_int == 2:logger_level = logging.INFOelif log_level_int == 3:logger_level = logging.WARNINGelif log_level_int == 4:logger_level = logging.ERRORelif log_level_int == 5:logger_level = logging.CRITICALreturn logger_leveldef __add_handlers(self):if self._is_add_stream_handler:self.__add_stream_handler()if all([self._log_path, self._log_filename]):self.__add_file_handler()def __add_stream_handler(self):"""日志显示到控制台"""stream_handler = logging.StreamHandler()stream_handler.setLevel(self._logger_level)stream_handler.setFormatter(self._formatter)self.logger.addHandler(stream_handler)def __add_file_handler(self):"""日志写入日志文件"""if not os.path.exists(self._log_path):os.makedirs(self._log_path)log_file = os.path.join(self._log_path, self._log_filename)os_name = os.nameprint os_namerotate_file_handler = Noneif os_name == 'nt':# windows下用这个，非进程安全rotate_file_handler = RotatingFileHandler(log_file, mode="a", maxBytes=10 * 1024 * 1024, backupCount=10, encoding="utf-8")if os_name == 'posix':# linux下可以使用ConcurrentRotatingFileHandler，进程安全的日志方式rotate_file_handler = ConcurrentRotatingFileHandler(log_file, mode="a", maxBytes=10 * 1024 * 1024, backupCount=10, encoding="utf-8")rotate_file_handler.setLevel(logging.DEBUG)rotate_file_handler.setFormatter(self._formatter)self.logger.addHandler(rotate_file_handler)def test_func():"""测试日志打印，单独运行test_func函数是不会打印出这条日志，没有添加handler"""logger = logging.getLogger('test')logger.info(u'需要被打印的日志')if __name__ == "__main__":test_log = Log('test', 1, log_path='./logs', log_filename='test.log')test_func()test_log.info(u'添加一条日志')import requestsrequest_log = Log('urllib3.connectionpool', 1)  # 测试三方包的日志resp = requests.get('https://www.baidu.com')