精简版logging

# coding=utf-8
import logging
import time
import os
import logging.handlers
import re


def logger(schedule_name, console_print=True, logging_level=None,console_debug_level=logging.INFO):
    """
    :param schedule_name: the log scheduler  name 
    :param console_print: if True console print log ,False not .
    :param logging_level: file logging generate levels support: ["DEBUG","INFO","WARNING","ERROR"].
    :param console_debug_level: it can be str also ,for exp: console_debug_level="INFO",default set logging.INFO.
    :return: logging handlers
    """
    log_fmt = "[%(levelname)s]%(asctime)s line %(lineno)d :\n%(message)s"
    c_fmt = "[%(levelname)s]%(asctime)s %(filename)s.%(funcName)s():line %(lineno)d :\n%(message)s"
    date_format = "%Y-%m-%d %H:%M:%S %a"
    # 设置控制台输出level
    logging.basicConfig(level=console_debug_level,
                        format=c_fmt,
                        datefmt=date_format,

                        )
    levels = []
    if isinstance(logging_level, list):
        for  level in logging_level:
            if level in ["DEBUG","INFO","WARNING","ERROR"]:
                levels.append(level)
        if levels:
            stamp = "dailylog.log"
            logsdir = os.path.join(os.getcwd(), "logs")
            if os.path.exists(logsdir):
                for p in levels:
                    if os.path.exists(os.path.join(logsdir, p)):
                        pass
                    else:
                        os.mkdir(os.path.join(logsdir, p))
            else:
                os.mkdir(logsdir)
                for p in levels:
                    os.mkdir(os.path.join(logsdir, p))

            logging_level_path = {}
            for i in levels:
                filename = os.path.join(logsdir, i, stamp)
                logging_level_path[i] = filename
            logger = logging.getLogger(schedule_name)
            print(logging_level_path)
            for k, v in logging_level_path.items():
                handler = logging.handlers.TimedRotatingFileHandler(filename=v, when='MIDNIGHT', interval=1,
                                                                    backupCount=4,encoding="utf-8")
                handler.suffix = "%Y-%m-%d.log"
                handler.extMatch = r"^\d{4}-\d{2}-\d{2}.log$"
                handler.extMatch = re.compile(handler.extMatch)
                h_fmt = logging.Formatter(log_fmt)
                handler.setFormatter(h_fmt)
                if k in ["DEBUG","INFO","WARNING","ERROR"]:
                   handler.setLevel(k)
                logger.addHandler(handler)
            logger.propagate = console_print
            return logger
        else:
            raise TypeError('logging_level support in:["DEBUG","INFO","WARNING","ERROR"]')
    else:
        raise NameError("logging_level expect list but get %s"%(type(logging_level).__name__))

  .

猜你喜欢

转载自www.cnblogs.com/SunshineKimi/p/11823729.html