Python–logging模組不同級別寫入到不同檔案

別來無恙-發表於2018-12-26
Python

將不同級別的logging 日誌資訊寫入到不同檔案

# -*- coding: utf-8 -*-
import os
import time
import logging
import inspect
from logging.handlers import RotatingFileHandler


dir = os.path.dirname(__file__)
dir_time = time.strftime(`%Y-%m-%d`, time.localtime())

handlers = {logging.NOTSET: os.path.join(dir, `notset_%s.log`%dir_time),

            logging.DEBUG: os.path.join(dir, `debug_%s.log`%dir_time),

            logging.INFO: os.path.join(dir, `info_%s.log`%dir_time),

            logging.WARNING: os.path.join(dir, `warning_%s.log`%dir_time),

            logging.ERROR: os.path.join(dir, `error_%s.log`%dir_time),

            logging.CRITICAL: os.path.join(dir, `critical_%s.log`%dir_time),
            }


def createHandlers():
    logLevels = handlers.keys()

    for level in logLevels:
        path = os.path.abspath(handlers[level])
        handlers[level] = RotatingFileHandler(path, maxBytes=10000, backupCount=2, encoding=`utf-8`)

# 載入模組時建立全域性變數

createHandlers()


class TNLog(object):

    def printfNow(self):
        return time.strftime(`%Y-%m-%d %H:%M:%S`, time.localtime())

    def __init__(self, level=logging.NOTSET):
        self.__loggers = {}

        logLevels = handlers.keys()

        for level in logLevels:
            logger = logging.getLogger(str(level))

            # 如果不指定level，獲得的handler似乎是同一個handler?

            logger.addHandler(handlers[level])

            logger.setLevel(level)

            self.__loggers.update({level: logger})

    def getLogMessage(self, level, message):
        frame, filename, lineNo, functionName, code, unknowField = inspect.stack()[2]

        ```日誌格式：[時間] [型別] [記錄程式碼] 資訊```

        return "[%s] [%s] [%s - %s - %s] %s" % (self.printfNow(), level, filename, lineNo, functionName, message)

    def info(self, message):
        message = self.getLogMessage("info", message)

        self.__loggers[logging.INFO].info(message)

    def error(self, message):
        message = self.getLogMessage("error", message)

        self.__loggers[logging.ERROR].error(message)

    def warning(self, message):
        message = self.getLogMessage("warning", message)

        self.__loggers[logging.WARNING].warning(message)

    def debug(self, message):
        message = self.getLogMessage("debug", message)

        self.__loggers[logging.DEBUG].debug(message)

    def critical(self, message):
        message = self.getLogMessage("critical", message)

        self.__loggers[logging.CRITICAL].critical(message)


if __name__ == "__main__":
    logger = TNLog()

    logger.debug("debug")
    logger.info("info")
    logger.warning("warning")
    logger.error("error")
    logger.critical("critical")

 

相關文章