logging模块日志记录

# coding:utf-8
import logging, os
from time import strftime

now = strftime("%Y-%m-%d.%H.%M.%S.")


class Log(object):
    def __init__(self):
        # 文件的命名
        self.logName = os.path.join(os.path.dirname(__file__), now + 'test.log')
        self.logger = logging.getLogger()
        self.logger.setLevel(logging.DEBUG)
        # 日志输出格式
        self.formatter = logging.Formatter(
            fmt="%(asctime)s  %(levelname)s :%(message)s",
            datefmt="%Y-%m-%d  %H:%M:%S"
        )

    def __console(self, level, message):
        # 创建一个 FileHandler，用于写到本地
        # fh = logging.FileHandler(self.logName, 'a') # 追加模式 这个是python2的
        fh = logging.FileHandler(self.logName, 'a', encoding='utf-8')  # 这个是 python3的
        fh.setLevel(logging.DEBUG)
        fh.setFormatter(self.formatter)
        self.logger.addHandler(fh)
        # 创建一个 StreamHandler,用于输出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.DEBUG)
        ch.setFormatter(self.formatter)
        self.logger.addHandler(ch)
        if level == 'info':
            self.logger.info(message)
        elif level == 'debug':
            self.logger.debug(message)
        elif level == 'warning':
            self.logger.warning(message)
        elif level == 'error':
            self.logger.error(message)
        # 这两行代码是为了避免日志输出重复问题
        self.logger.removeHandler(ch)
        self.logger.removeHandler(fh)
        # 关闭打开的文件
        fh.close()

    def debug(self, message):
        self.__console('debug', message)

    def info(self, message):
        self.__console('info', message)

    def warning(self, message):
        self.__console('warning', message)

    def error(self, message):
        self.__console('error', message)